• Home
  • Raw
  • Download

Lines Matching refs:group_output_channels

52   const size_t group_output_channels = state.range(11);  in xnnpack_convolution_q8()  local
59 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_convolution_q8()
72 …std::vector<uint8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_… in xnnpack_convolution_q8()
74 std::vector<int32_t> bias(groups * group_output_channels); in xnnpack_convolution_q8()
100 groups, group_input_channels, group_output_channels, in xnnpack_convolution_q8()
153 groups * group_input_channels * group_output_channels * in xnnpack_convolution_q8()
170 const size_t group_output_channels = state.range(11); in xnnpack_convolution_f32() local
176 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_convolution_f32()
189 …std::vector<float> kernel(groups * group_output_channels * kernel_height * kernel_width * group_in… in xnnpack_convolution_f32()
191 std::vector<float> bias(groups * group_output_channels); in xnnpack_convolution_f32()
217 groups, group_input_channels, group_output_channels, in xnnpack_convolution_f32()
267 groups * group_input_channels * group_output_channels * in xnnpack_convolution_f32()
285 const size_t group_output_channels = state.range(11); in tflite_convolution_f32() local
317 …std::vector<float> kernel(groups * group_output_channels * kernel_height * kernel_width * group_in… in tflite_convolution_f32()
319 std::vector<float> bias(groups * group_output_channels); in tflite_convolution_f32()
340 static_cast<int32_t>(group_output_channels), in tflite_convolution_f32()
364 static_cast<int32_t>(groups * group_output_channels) in tflite_convolution_f32()
367 static_cast<int32_t>(group_output_channels), in tflite_convolution_f32()
373 static_cast<int32_t>(groups * group_output_channels) in tflite_convolution_f32()
474 groups * group_input_channels * group_output_channels * in tflite_convolution_f32()
499 const size_t group_output_channels = state.range(11); in compare_with_convolution_f32_reference_output() local
511 …assert(kernel_size == group_output_channels * kernel_height * kernel_width * groups * group_input_… in compare_with_convolution_f32_reference_output()
513 assert(bias_size == groups * group_output_channels); in compare_with_convolution_f32_reference_output()
515 assert(output_size == batch_size * output_height * output_width * groups * group_output_channels); in compare_with_convolution_f32_reference_output()
522 for (size_t oc = 0; oc < group_output_channels; oc++) { in compare_with_convolution_f32_reference_output()
523 …t_ref[(((i * output_height + oy) * output_width + ox) * groups + g) * group_output_channels + oc] = in compare_with_convolution_f32_reference_output()
524 bias[g * group_output_channels + oc]; in compare_with_convolution_f32_reference_output()
540 for (size_t oc = 0; oc < group_output_channels; oc++) { in compare_with_convolution_f32_reference_output()
542 …_ref[(((i * output_height + oy) * output_width + ox) * groups + g) * group_output_channels + oc] += in compare_with_convolution_f32_reference_output()
561 for (size_t c = 0; c < group_output_channels; c++) { in compare_with_convolution_f32_reference_output()
562 …ze_t idx = (((i * output_height + y) * output_width + x) * groups + g) * group_output_channels + c; in compare_with_convolution_f32_reference_output()
593 const size_t group_output_channels = state.range(11); in armcl_convolution_f32() local
633 /* O */ group_output_channels); in armcl_convolution_f32()
643 arm_compute::TensorShape bias_shape(groups * group_output_channels); in armcl_convolution_f32()
655 /* C */ groups * group_output_channels, in armcl_convolution_f32()
715 pad_stride_info, group_output_channels, activation_info); in armcl_convolution_f32()
719 pad_stride_info, group_output_channels, activation_info)) in armcl_convolution_f32()
729 pad_stride_info, group_output_channels, activation_info); in armcl_convolution_f32()
733 pad_stride_info, group_output_channels, activation_info)) in armcl_convolution_f32()
797 groups * group_input_channels * group_output_channels * in armcl_convolution_f32()