Home
last modified time | relevance | path

Searched defs:kr_block_offset (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Dpack.h44 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8gemm_wdq() local
94 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8gemm_wrq() local
107 for (size_t kr_block_offset = 0; kr_block_offset < (kr - kr_block_size); in pytorch_pack_q8gemm_wrq() local
125 for (size_t kr_block_offset = 0; kr_block_offset < kr; in pytorch_pack_q8gemm_wrq() local
168 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8conv_wdq() local
217 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8conv_wrq() local
230 for (size_t kr_block_offset = 0; kr_block_offset < (kr - kr_block_size); in pytorch_pack_q8conv_wrq() local
246 for (size_t kr_block_offset = 0; kr_block_offset < kr; in pytorch_pack_q8conv_wrq() local
290 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8deconv_wdq() local
339 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in pytorch_pack_q8deconv_wrq() local
[all …]
/aosp_15_r20/external/XNNPACK/src/
H A Dpacking.c49 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f32_gemm_goi_w() local
96 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f16_gemm_goi_w() local
143 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f32_to_f16_gemm_goi_w() local
201 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_qu8_gemm_goi_w() local
261 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_qs8_gemm_goi_w() local
321 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_qs8_gemm_xw_goi_w() local
368 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f32_gemm_io_w() local
406 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f16_gemm_io_w() local
444 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_f32_to_f16_gemm_io_w() local
493 for (size_t kr_block_offset = 0; kr_block_offset < kr; kr_block_offset++) { in xnn_pack_qu8_gemm_io_w() local
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dfc-unpack.cc39 for (size_t kr_block_offset = 0; kr_block_offset < kr_block_size; in unpackWeights() local
45 for (size_t kr_block_offset = 0; kr_block_offset < (kr - kr_block_size); in unpackWeights() local
57 for (size_t kr_block_offset = 0; kr_block_offset < kr; in unpackWeights() local