Searched refs:kc_stride (Results 1 – 6 of 6) sorted by relevance
| /external/XNNPACK/bench/ |
| D | f16-gemm.cc | 44 const size_t kc_stride = benchmark::utils::RoundUp(kc, kr); 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-im2col-gemm.cc | 61 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); 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 | 51 const size_t kc_stride = benchmark::utils::RoundUp(kc, kr); in GEMMBenchmark() local
65 const size_t w_elements = kc_stride * nc_stride + nc_stride * sizeof(int32_t) / sizeof(uint8_t); in GEMMBenchmark()
98 … w.data() + (w_elements * buffer_index + n * (kc_stride + sizeof(int32_t))) / sizeof(uint8_t), in GEMMBenchmark()
|
| D | f32-igemm.cc | 68 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in IGEMMBenchmark() local
79 const size_t w_elements = (kernel_size * kc_stride + 1) * nc_stride; in IGEMMBenchmark()
141 w.data() + buffer_index * w_elements + n * (kc_stride * kernel_size + 1), in IGEMMBenchmark()
|
| D | f32-gemm.cc | 54 const size_t kc_stride = benchmark::utils::RoundUp(kc, kr); 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()
|
| /external/XNNPACK/src/q8-gemm/ |
| D | 2x4c8-sse2.c | 61 const size_t kc_stride = round_up_po2(kc, 8); in xnn_q8_gemm_ukernel_2x4c8__sse2() local
176 a0 = (const uint8_t*) ((uintptr_t) a0 - kc_stride); in xnn_q8_gemm_ukernel_2x4c8__sse2()
177 a1 = (const uint8_t*) ((uintptr_t) a1 - kc_stride); in xnn_q8_gemm_ukernel_2x4c8__sse2()
|