| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
| A D | depth_to_space.h | 56 for (int out_d = 0; out_d < output_depth; ++out_d) { in DepthToSpace() local
58 out_d + ((out_h % block_size) * block_size + out_w % block_size) * in DepthToSpace()
67 Offset(output_shape, out_b, out_h, out_w, out_d); in DepthToSpace()
|
| A D | conv3d.h | 55 for (int out_d = 0; out_d < output_depth; ++out_d) { in Conv3D() local
56 const int in_d_origin = (out_d * params.stride_depth) - pad_depth; in Conv3D()
99 output_data[Offset(output_shape, batch, out_d, out_y, out_x, in Conv3D()
|
| A D | pad.h | 95 for (int out_d = 0; out_d < output_depth; ++out_d) { in PadImpl() local
104 out_d < left_d_padding || in PadImpl()
105 out_d >= output_depth - right_d_padding) { in PadImpl()
|
| A D | reduce.h | 247 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean() local
251 value += input_data[Offset(input_shape, out_b, in_h, in_w, out_d)]; in Mean()
254 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in Mean()
303 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean() local
307 acc += input_data[Offset(input_shape, out_b, in_h, in_w, out_d)]; in Mean()
313 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in Mean()
|
| A D | space_to_depth.h | 57 const int out_d = in SpaceToDepth() local
66 Offset(output_shape, out_b, out_h, out_w, out_d); in SpaceToDepth()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| A D | mean.h | 55 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean() local
59 acc += input_data[Offset(input_shape, out_b, in_h, in_w, out_d)] - in Mean()
68 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in Mean()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| A D | mean.h | 56 int out_d = start_depth; in MeanImpl() local
59 for (; out_d <= end_depth - 16; out_d += 16) { in MeanImpl()
68 input_data + Offset(input_shape, out_b, in_h, in_w, out_d); in MeanImpl()
121 output_data + Offset(output_shape, out_b, 0, 0, out_d); in MeanImpl()
126 for (; out_d < end_depth; ++out_d) { in MeanImpl()
130 acc += input_data[Offset(input_shape, out_b, in_h, in_w, out_d)]; in MeanImpl()
137 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in MeanImpl()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
| A D | im2col_utils.h | 453 for (int out_d = 0; out_d < output_depth; ++out_d) { in DilatedIm2col3D() local
454 const int in_d_origin = (out_d * params.stride_depth) - pad_depth; in DilatedIm2col3D()
460 Offset(row_shape, 0, batch, out_d, out_y, out_x); in DilatedIm2col3D()
|
| A D | optimized_ops.h | 1000 int out_d = start_depth; in MeanImpl() local
1003 for (; out_d <= end_depth - 16; out_d += 16) { in MeanImpl()
1012 input_data + Offset(input_shape, out_b, in_h, in_w, out_d); in MeanImpl()
1069 output_data + Offset(output_shape, out_b, 0, 0, out_d); in MeanImpl()
1074 for (; out_d < end_depth; ++out_d) { in MeanImpl()
1078 acc += input_data[Offset(input_shape, out_b, in_h, in_w, out_d)]; in MeanImpl()
1085 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in MeanImpl()
|