Searched defs:nr_block_size (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/ |
| D | packing.c | 39 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_goi_w() local
86 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_goi_w() local
133 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_to_f16_gemm_goi_w() local
182 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qu8_gemm_goi_w() local
242 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qs8_gemm_goi_w() local
302 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qs8_gemm_xw_goi_w() local
358 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_io_w() local
396 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_io_w() local
434 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_to_f16_gemm_io_w() local
474 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qu8_gemm_io_w() local
[all …]
|
| D | operator-run.c | 302 size_t nr_block_size) in xnn_compute_grouped_gemm()
326 size_t nr_block_size) in xnn_compute_gemm()
369 size_t nr_block_size) in xnn_compute_grouped_batch_igemm()
395 size_t nr_block_size) in xnn_compute_grouped_igemm()
421 size_t nr_block_size) in xnn_compute_batch_igemm()
446 size_t nr_block_size) in xnn_compute_igemm()
1185 size_t nr_block_size) in xnn_compute_hmp_grouped_gemm()
1210 size_t nr_block_size) in xnn_compute_hmp_gemm()
1236 size_t nr_block_size) in xnn_compute_hmp_grouped_batch_igemm()
1263 size_t nr_block_size) in xnn_compute_hmp_grouped_igemm()
[all …]
|