Lines Matching refs:group_input_channels

43   const size_t group_input_channels = state.range(8);  in Im2ColGEMMBenchmark()  local
59   const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr);  in Im2ColGEMMBenchmark()
61   std::vector<float> a(input_height * input_width * group_input_channels);  in Im2ColGEMMBenchmark()
63   std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels);  in Im2ColGEMMBenchmark()
76   xnn_pack_f32_gemm_goi_w(1 /* groups */, group_output_channels, group_input_channels * kernel_size,  in Im2ColGEMMBenchmark()
82 …std::vector<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_c…  in Im2ColGEMMBenchmark()
105         group_input_channels * sizeof(float) /* input channels */,  in Im2ColGEMMBenchmark()
106         group_input_channels * sizeof(float) /* input stride */,  in Im2ColGEMMBenchmark()
116           mb, nb, kernel_size * group_input_channels * sizeof(float),  in Im2ColGEMMBenchmark()
117 …inputData + m * kernel_size * group_input_channels, kernel_size * group_input_channels * sizeof(fl…  in Im2ColGEMMBenchmark()
133       group_input_channels * group_output_channels *  in Im2ColGEMMBenchmark()