Searched refs:TF_LITE_SPACE_TO_BATCH_ND (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | space_to_batch_nd.cc | 116 #define TF_LITE_SPACE_TO_BATCH_ND(type, scalar, pad_value) \ in Eval() macro
130 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, float, 0); in Eval()
132 TF_LITE_SPACE_TO_BATCH_ND(optimized_ops, float, 0); in Eval()
137 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, uint8_t, in Eval()
140 TF_LITE_SPACE_TO_BATCH_ND(optimized_ops, uint8_t, in Eval()
146 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, int8_t, in Eval()
149 TF_LITE_SPACE_TO_BATCH_ND(optimized_ops, int8_t, in Eval()
155 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, int32_t, 0); in Eval()
157 TF_LITE_SPACE_TO_BATCH_ND(optimized_ops, int32_t, 0); in Eval()
162 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, int64_t, 0); in Eval()
[all …]
|