Searched refs:nc_stride (Results 1 – 5 of 5) sorted by relevance
| /external/XNNPACK/bench/ |
| D | f16-gemm.cc | 43 const size_t nc_stride = benchmark::utils::RoundUp(nc, nr); in GEMMBenchmark() local
58 const size_t w_elements = nc_stride * kc_stride + nc_stride; in GEMMBenchmark()
91 w.data() + (nc_stride * buffer_index + n) * (kc_stride + 1), in GEMMBenchmark()
|
| D | f32-gemm.cc | 53 const size_t nc_stride = benchmark::utils::RoundUp(nc, nr); in GEMMBenchmark() local
67 const size_t w_elements = nc_stride * kc_stride + nc_stride; in GEMMBenchmark()
98 w.data() + buffer_index * nc_stride * (kc_stride + 1), in GEMMBenchmark()
127 const size_t nc_stride = benchmark::utils::RoundUp(nc, nr); in PPMM1PBenchmark() local
142 const size_t w_elements = nc_stride * kc + nc_stride; in PPMM1PBenchmark()
174 w.data() + nc_stride * buffer_index * (kc + 1), in PPMM1PBenchmark()
204 const size_t nc_stride = benchmark::utils::RoundUp(nc, nr); in PPMM2PBenchmark() local
219 const size_t w_elements = nc_stride * kc + nc_stride; in PPMM2PBenchmark()
254 w.data() + nc_stride * buffer_index * (kc + 1), in PPMM2PBenchmark()
|
| D | f32-im2col-gemm.cc | 60 const size_t nc_stride = benchmark::utils::RoundUp<size_t>(group_output_channels, nr); in Im2ColGEMMBenchmark() local
70 const size_t w_elements = (kernel_size * kc_stride + 1) * nc_stride; in Im2ColGEMMBenchmark()
120 w.data() + (buffer_index * nc_stride + n) * (kernel_size * kc_stride + 1), in Im2ColGEMMBenchmark()
|
| D | q8-gemm.cc | 50 const size_t nc_stride = benchmark::utils::RoundUp(nc, nr); in GEMMBenchmark() local
65 const size_t w_elements = kc_stride * nc_stride + nc_stride * sizeof(int32_t) / sizeof(uint8_t); in GEMMBenchmark()
|
| D | f32-igemm.cc | 67 const size_t nc_stride = benchmark::utils::RoundUp<size_t>(group_output_channels, nr); in IGEMMBenchmark() local
79 const size_t w_elements = (kernel_size * kc_stride + 1) * nc_stride; in IGEMMBenchmark()
|