Searched defs:nr_block_size (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/ |
| D | packing.c | 33 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_goi_w() local
87 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_goi_w() local
141 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qu8_gemm_goi_w() local
195 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qs8_gemm_goi_w() local
249 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qs8_gemm_xw_goi_w() local
302 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_io_w() local
349 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_io_w() local
396 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_qu8_gemm_io_w() local
447 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_conv_goki_w() local
504 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_conv_goki_w() local
[all …]
|
| D | operator-run.c | 30 size_t nr_block_size) in xnn_compute_grouped_gemm()
54 size_t nr_block_size) in xnn_compute_gemm()
97 size_t nr_block_size) in xnn_compute_grouped_batch_igemm()
123 size_t nr_block_size) in xnn_compute_grouped_igemm()
149 size_t nr_block_size) in xnn_compute_batch_igemm()
174 size_t nr_block_size) in xnn_compute_igemm()
906 size_t nr_block_size) in xnn_compute_hmp_grouped_gemm()
931 size_t nr_block_size) in xnn_compute_hmp_gemm()
957 size_t nr_block_size) in xnn_compute_hmp_grouped_batch_igemm()
984 size_t nr_block_size) in xnn_compute_hmp_grouped_igemm()
[all …]
|