Lines Matching refs:group_output_channels

46   const size_t group_output_channels = state.range(9);  in Im2ColGEMMBenchmark()  local
60   const size_t nc_stride = benchmark::utils::RoundUp<size_t>(group_output_channels, nr);  in Im2ColGEMMBenchmark()
65   std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels);  in Im2ColGEMMBenchmark()
67   std::vector<float> b(group_output_channels);  in Im2ColGEMMBenchmark()
71   const size_t c_elements = output_size * group_output_channels;  in Im2ColGEMMBenchmark()
78   xnn_pack_f32_gemm_goi_w(1 /* groups */, group_output_channels, group_input_channels * kernel_size,  in Im2ColGEMMBenchmark()
84 …tor<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_channels);  in Im2ColGEMMBenchmark()
115       for (uint32_t n = 0; n < group_output_channels; n += nr) {  in Im2ColGEMMBenchmark()
116         const uint32_t nb = min(group_output_channels - n, nr);  in Im2ColGEMMBenchmark()
121 …c.data() + (buffer_index * output_size + m) * group_output_channels + n, group_output_channels * s…  in Im2ColGEMMBenchmark()
131       group_input_channels * group_output_channels *  in Im2ColGEMMBenchmark()