Home
last modified time | relevance | path

Searched defs:nr_block_start (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/
H A Dpacking.c38 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemm_goi_w() local
85 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f16_gemm_goi_w() local
132 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_to_f16_gemm_goi_w() local
181 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_qu8_gemm_goi_w() local
241 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_qs8_gemm_goi_w() local
301 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_qs8_gemm_xw_goi_w() local
357 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemm_io_w() local
395 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f16_gemm_io_w() local
433 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_to_f16_gemm_io_w() local
473 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_qu8_gemm_io_w() local
[all …]
H A Doperator-run.c300 size_t nr_block_start, in xnn_compute_grouped_gemm()
324 size_t nr_block_start, in xnn_compute_gemm()
367 size_t nr_block_start, in xnn_compute_grouped_batch_igemm()
393 size_t nr_block_start, in xnn_compute_grouped_igemm()
419 size_t nr_block_start, in xnn_compute_batch_igemm()
444 size_t nr_block_start, in xnn_compute_igemm()
1183 size_t nr_block_start, in xnn_compute_hmp_grouped_gemm()
1208 size_t nr_block_start, in xnn_compute_hmp_gemm()
1234 size_t nr_block_start, in xnn_compute_hmp_grouped_batch_igemm()
1261 size_t nr_block_start, in xnn_compute_hmp_grouped_igemm()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Dpack.h29 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in pytorch_pack_q8gemm_wdq() local
83 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in pytorch_pack_q8gemm_wrq() local
151 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_q8conv_wdq() local
204 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_q8conv_wrq() local
273 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_q8deconv_wdq() local
326 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_q8deconv_wrq() local
546 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_swizzle_q8gemm_bdq() local
615 for (size_t nr_block_start = 0; nr_block_start < n; nr_block_start += nr) { in pytorch_pack_swizzle_q8gemm_brq() local
668 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in pytorch_pack_hgemm_w() local
700 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in pytorch_pack_sgemm_w() local
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Doperator-run.c44 size_t nr_block_start, in compute_q8gemm()
96 size_t nr_block_start, in compute_q8gemm_sparse_dq()
161 size_t nr_block_start, in compute_q8gemm_prepack_a_sparse()
184 size_t nr_block_start, in compute_q8gemm_prepacked_sparse_dq()
317 size_t nr_block_start, in compute_q8gemm_xzp()
372 size_t nr_block_start, in compute_q8conv()
H A Dconv-run.cc34 size_t nr_block_start, in compute_q8gemm_xzp()
85 size_t nr_block_start, in compute_q8gemm()
136 size_t nr_block_start, in compute_q8conv()
H A Dfc-unpack.cc28 for (size_t nr_block_start = 0; nr_block_start < output_channels_; nr_block_start += nr) { in unpackWeights() local
H A Dfc-run.cc25 size_t nr_block_start, in compute_q8gemm()
H A Dfc-dynamic-run.cc26 size_t nr_block_start, in compute_q8gemm_dq()
H A Ddeconv-run.cc49 size_t nr_block_start, in compute_q8conv()