Searched refs:TF_LITE_BATCH_TO_SPACE_ND (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd.cc | 127 #define TF_LITE_BATCH_TO_SPACE_ND(type, scalar) \ in Eval() macro
139 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, float); in Eval()
141 TF_LITE_BATCH_TO_SPACE_ND(optimized_ops, float); in Eval()
146 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, uint8_t); in Eval()
148 TF_LITE_BATCH_TO_SPACE_ND(optimized_ops, uint8_t); in Eval()
153 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, int8_t); in Eval()
155 TF_LITE_BATCH_TO_SPACE_ND(optimized_ops, int8_t); in Eval()
160 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, int32_t); in Eval()
162 TF_LITE_BATCH_TO_SPACE_ND(optimized_ops, int32_t); in Eval()
167 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, int64_t); in Eval()
[all …]
|