Home
last modified time | relevance | path

Searched refs:depth_multiplier (Results 1 – 19 of 19) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Ddepthwise_conv.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
58 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
65 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
66 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
136 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
155 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
162 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
163 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
221 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridPerChannel() local
239 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridPerChannel()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Ddepthwiseconv_uint8.h104 const int depth_multiplier = params.depth_multiplier; in Run() local
126 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
133 for (int m = 0; m < depth_multiplier; m++) { in Run()
134 const int oc = m + ic * depth_multiplier; in Run()
190 const int depth_multiplier = params.depth_multiplier; in RunPerChannel() local
213 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in RunPerChannel()
220 for (int m = 0; m < depth_multiplier; ++m) { in RunPerChannel()
221 const int output_channel = m + in_channel * depth_multiplier; in RunPerChannel()
A Ddepthwiseconv_float.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
53 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
60 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConv()
61 const int oc = m + ic * depth_multiplier; in DepthwiseConv()
A Dlegacy_reference_ops.h47 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
60 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
73 int pad_height, int depth_multiplier, in DepthwiseConv() argument
79 pad_height, depth_multiplier, output_activation_min, in DepthwiseConv()
104 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
108 depth_multiplier, output_data, output_dims); in DepthwiseConv()
131 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
151 int pad_height, int depth_multiplier, in DepthwiseConv() argument
181 stride_height, pad_width, pad_height, depth_multiplier, in DepthwiseConv()
193 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Ddepthwiseconv_float.h34 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
95 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
767 int depth_multiplier, int filter_width,
782 TFLITE_DCHECK_EQ(depth_multiplier, kFixedDepthMultiplier);
784 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
831 depth_multiplier,
843 const float* input_data, int pad_width, int depth_multiplier,
867 for (int m = 0; m < depth_multiplier; m++) {
913 const int depth_multiplier = params.depth_multiplier;
932 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
[all …]
A Ddepthwiseconv_uint8.h42 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1517 int pad_width, int depth_multiplier,
1533 TFLITE_DCHECK_EQ(depth_multiplier, kFixedDepthMultiplier);
1535 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
1593 int depth_multiplier, int filter_width, const uint8* filter_data,
1617 for (int m = 0; m < depth_multiplier; m++) {
1711 const int depth_multiplier = params.depth_multiplier;
1880 input_offset, pad_width, depth_multiplier, filter_width,
2043 const int depth_multiplier = params.depth_multiplier;
2056 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
[all …]
A Ddepthwiseconv_3x3_filter_common.h260 const int32 depth_multiplier = params.depth_multiplier;
268 (((input_depth % 8) == 0 && depth_multiplier == 1) ||
269 (input_depth == 1 && depth_multiplier > 1));
287 if (params.depth_multiplier == 1) {
443 int32 depth_multiplier, const RuntimeShape& output_shape,
454 filter_width == 3 && filter_height == 3 && depth_multiplier == 1 &&
A Dlegacy_optimized_ops.h155 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
173 int pad_height, int depth_multiplier, in DepthwiseConv() argument
179 pad_height, depth_multiplier, output_activation_min, in DepthwiseConv()
204 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
208 depth_multiplier, output_data, output_dims); in DepthwiseConv()
219 const int depth_multiplier = params.depth_multiplier; in LegacyDepthwiseConvWithRounding() local
232 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in LegacyDepthwiseConvWithRounding()
249 depth_multiplier, output_shape, output_shift)) { in LegacyDepthwiseConvWithRounding()
299 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
324 int pad_height, int depth_multiplier, in DepthwiseConv() argument
[all …]
A Ddepthwiseconv_uint8_3x3_filter.h5598 const int32 depth_multiplier = rt_params.depth_multiplier;
5631 TFLITE_DCHECK(params.output_depth == params.input_depth * depth_multiplier);
5632 TFLITE_DCHECK(depth_multiplier == 1);
12889 const int depth_multiplier = params.depth_multiplier;
12890 const bool has_depth_multiplication = depth_multiplier > 1;
12903 TFLITE_DCHECK_EQ(input_depth * depth_multiplier, output_depth);
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Ddepthwise_conv.h48 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
1427 int pad_width, int depth_multiplier,
1443 TFLITE_DCHECK_EQ(depth_multiplier, kFixedDepthMultiplier);
1445 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
1503 int depth_multiplier, int filter_width, const int8* filter_data,
1527 for (int m = 0; m < depth_multiplier; m++) {
1622 const int depth_multiplier = params.depth_multiplier;
1770 input_offset, pad_width, depth_multiplier, filter_width,
1803 const int depth_multiplier = params.depth_multiplier;
1813 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
[all …]
A Ddepthwise_conv_hybrid.h53 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridGeneral() local
88 depth_multiplier == FIXED_DEPTH_MULTIPLIER) { \ in DepthwiseConvHybridGeneral()
199 -input_offset, pad_width, depth_multiplier, filter_width, in DepthwiseConvHybridGeneral()
268 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridWithRounding() local
278 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridWithRounding()
296 depth_multiplier, output_shape, 0, nullptr)) { in DepthwiseConvHybridWithRounding()
A Ddepthwise_conv_hybrid_3x3_filter.h3058 const int32 depth_multiplier = rt_params.depth_multiplier;
3082 TFLITE_DCHECK(params.output_depth == params.input_depth * depth_multiplier);
3083 TFLITE_DCHECK(depth_multiplier == 1);
A Ddepthwise_conv_3x3_filter.h2892 const int32 depth_multiplier = rt_params.depth_multiplier;
2921 TFLITE_DCHECK(params.output_depth == params.input_depth * depth_multiplier);
2922 TFLITE_DCHECK(depth_multiplier == 1);
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Ddepthwise_conv_common.cc54 op_params.depth_multiplier = params.depth_multiplier; in DepthwiseConvParamsFloat()
75 op_params.depth_multiplier = params.depth_multiplier; in DepthwiseConvParamsQuantized()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Ddepthwise_conv.cc174 dw_conv_params.ch_mult = params.depth_multiplier; in EvalQuantizedPerChannel()
271 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dbuiltin_op_data.h127 int depth_multiplier; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/
A Dflatbuffer_conversions.cc1176 params->depth_multiplier = schema_params->depth_multiplier(); in ParseDepthwiseConv2D()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dtypes.h899 int16_t depth_multiplier; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/
A Dschema_generated.h4314 int32_t depth_multiplier;
4322 depth_multiplier(0),
4349 int32_t depth_multiplier() const {
4389 void add_depth_multiplier(int32_t depth_multiplier) {
4390 fbb_.AddElement<int32_t>(DepthwiseConv2DOptions::VT_DEPTH_MULTIPLIER, depth_multiplier, 0);
4418 int32_t depth_multiplier = 0,
4425 builder_.add_depth_multiplier(depth_multiplier);
12209 { auto _e = depth_multiplier(); _o->depth_multiplier = _e; }
12226 auto _depth_multiplier = _o->depth_multiplier;

Completed in 115 milliseconds