Home
last modified time | relevance | path

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

/aosp_15_r20/external/XNNPACK/src/
H A Dpacking.c39 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 …]
H A Doperator-run.c302 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 …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Dpack.h30 const size_t nr_block_size = min(nc - nr_block_start, nr); in pytorch_pack_q8gemm_wdq() local
84 const size_t nr_block_size = min(nc - nr_block_start, nr); in pytorch_pack_q8gemm_wrq() local
152 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_q8conv_wdq() local
205 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_q8conv_wrq() local
274 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_q8deconv_wdq() local
327 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_q8deconv_wrq() local
547 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_swizzle_q8gemm_bdq() local
616 const size_t nr_block_size = min(n - nr_block_start, nr); in pytorch_pack_swizzle_q8gemm_brq() local
669 const size_t nr_block_size = min(nc - nr_block_start, nr); in pytorch_pack_hgemm_w() local
701 const size_t nr_block_size = min(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.c48 size_t nr_block_size) { in compute_q8gemm()
100 size_t nr_block_size) { in compute_q8gemm_sparse_dq()
165 size_t nr_block_size) { in compute_q8gemm_prepack_a_sparse()
188 size_t nr_block_size) { in compute_q8gemm_prepacked_sparse_dq()
321 size_t nr_block_size) { in compute_q8gemm_xzp()
376 size_t nr_block_size) { in compute_q8conv()
H A Dconv-run.cc38 size_t nr_block_size) { in compute_q8gemm_xzp()
89 size_t nr_block_size) { in compute_q8gemm()
140 size_t nr_block_size) { in compute_q8conv()
H A Dfc-unpack.cc29 const size_t nr_block_size = min(output_channels_ - nr_block_start, nr); in unpackWeights() local
H A Dfc-run.cc29 size_t nr_block_size) in compute_q8gemm()
H A Dfc-dynamic-run.cc30 size_t nr_block_size) { in compute_q8gemm_dq()
H A Ddeconv-run.cc53 size_t nr_block_size) { in compute_q8conv()