Lines Matching refs:group_input_channels

52   const size_t group_input_channels = state.range(10);  in xnnpack_convolution_qu8()  local
61 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_qu8()
73 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_qu8()
97 groups, group_input_channels, group_output_channels, in xnnpack_convolution_qu8()
154 groups * group_input_channels * group_output_channels * in xnnpack_convolution_qu8()
172 const size_t group_input_channels = state.range(10); in xnnpack_convolution_qs8() local
182 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_qs8()
194 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_qs8()
218 groups, group_input_channels, group_output_channels, in xnnpack_convolution_qs8()
274 groups * group_input_channels * group_output_channels * in xnnpack_convolution_qs8()
295 const size_t group_input_channels = state.range(10); in xnnpack_convolution_f16() local
304 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_f16()
316 …16_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_f16()
340 groups, group_input_channels, group_output_channels, in xnnpack_convolution_f16()
394 groups * group_input_channels * group_output_channels * in xnnpack_convolution_f16()
411 const size_t group_input_channels = state.range(10); in xnnpack_convolution_f32() local
419 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_f32()
431 …loat> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_f32()
455 groups, group_input_channels, group_output_channels, in xnnpack_convolution_f32()
509 groups * group_input_channels * group_output_channels * in xnnpack_convolution_f32()
526 const size_t group_input_channels = state.range(10); in tflite_convolution_f32() local
531 if (group_input_channels == 1) { in tflite_convolution_f32()
559 …loat> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in tflite_convolution_f32()
600 static_cast<int32_t>(groups * group_input_channels) in tflite_convolution_f32()
612 static_cast<int32_t>(groups * group_input_channels) in tflite_convolution_f32()
695 …interpreter->typed_tensor<float>(0) + batch_size * groups * group_input_channels * input_height * … in tflite_convolution_f32()
703 batch_size * groups * group_input_channels * input_height * input_width * sizeof(float)); in tflite_convolution_f32()
720 groups * group_input_channels * group_output_channels * in tflite_convolution_f32()
744 const size_t group_input_channels = state.range(10); in compare_with_convolution_f32_reference_output() local
751 const size_t input_pixel_stride = groups * group_input_channels; in compare_with_convolution_f32_reference_output()
755 assert(input_size == batch_size * input_height * input_width * groups * group_input_channels); in compare_with_convolution_f32_reference_output()
757 …rnel_size == group_output_channels * kernel_height * kernel_width * groups * group_input_channels); in compare_with_convolution_f32_reference_output()
787 for (size_t ic = 0; ic < group_input_channels; ic++) { in compare_with_convolution_f32_reference_output()
789 …(i * input_height + iy) * input_width + ix) * input_pixel_stride + g * group_input_channels + ic] * in compare_with_convolution_f32_reference_output()
790 …kernel[(((oc * kernel_height + ky) * kernel_width + kx) * groups + g) * group_input_channels + ic]; in compare_with_convolution_f32_reference_output()
838 const size_t group_input_channels = state.range(10); in armcl_convolution_f32() local
861 /* C */ groups * group_input_channels, in armcl_convolution_f32()
876 /* I */ groups * group_input_channels, in armcl_convolution_f32()
939 if (group_input_channels == 1) { in armcl_convolution_f32()
1012 batch_size * groups * group_input_channels * input_height * input_width * sizeof(float)); in armcl_convolution_f32()
1047 groups * group_input_channels * group_output_channels * in armcl_convolution_f32()