Home
last modified time | relevance | path

Searched refs:output_batch (Results 1 – 7 of 7) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Ddepth_to_space.h44 const int output_batch = output_shape.Dims(0); in DepthToSpace() local
51 TFLITE_DCHECK_EQ(input_batch, output_batch); in DepthToSpace()
53 for (int out_b = 0; out_b < output_batch; ++out_b) { in DepthToSpace()
A Dspace_to_depth.h44 const int output_batch = output_shape.Dims(0); in SpaceToDepth() local
51 TFLITE_DCHECK_EQ(input_batch, output_batch); in SpaceToDepth()
A Dpad.h69 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
91 for (int out_b = 0; out_b < output_batch; ++out_b) { in PadImpl()
97 out_b >= output_batch - right_b_padding || in PadImpl()
A Dreduce.h232 const int output_batch = output_shape.Dims(0); in Mean() local
246 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
276 const int output_batch = output_shape.Dims(0); in Mean() local
302 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Dmean.h37 const int output_batch = output_shape.Dims(0); in Mean() local
54 for (int out_b = 0; out_b < output_batch; ++out_b) { in Mean()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Dmean.h35 const int output_batch = output_shape.Dims(0); in MeanImpl() local
55 for (int out_b = 0; out_b < output_batch; ++out_b) { in MeanImpl()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Doptimized_ops.h978 const int output_batch = output_shape.Dims(0); in MeanImpl() local
999 for (int out_b = 0; out_b < output_batch; ++out_b) { in MeanImpl()
4648 const int output_batch = ext_output_shape.Dims(0); in PadImpl() local
4674 for (int out_b = left_b_padding; out_b < output_batch - right_b_padding; in PadImpl()
4744 output_data + Offset(ext_output_shape, output_batch - right_b_padding, in PadImpl()

Completed in 29 milliseconds