Searched defs:kr_block_start (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | pack.h | 39 for (size_t kr_block_start = 0; kr_block_start < kc; kr_block_start += kr) { in pytorch_pack_q8gemm_wdq() local
90 for (size_t kr_block_start = 0; kr_block_start < kc; kr_block_start += kr) { in pytorch_pack_q8gemm_wrq() local
162 for (size_t kr_block_start = 0; kr_block_start < kc; in pytorch_pack_q8conv_wdq() local
212 for (size_t kr_block_start = 0; kr_block_start < kc; in pytorch_pack_q8conv_wrq() local
284 for (size_t kr_block_start = 0; kr_block_start < kc; in pytorch_pack_q8deconv_wdq() local
334 for (size_t kr_block_start = 0; kr_block_start < kc; in pytorch_pack_q8deconv_wrq() local
557 for (size_t kr_block_start = 0; kr_block_start < (kc & -sr); in pytorch_pack_swizzle_q8gemm_bdq() local
577 for (size_t kr_block_start = (kc & -sr); kr_block_start < kc; in pytorch_pack_swizzle_q8gemm_bdq() local
624 for (size_t kr_block_start = 0; kr_block_start < (kc & -sr); in pytorch_pack_swizzle_q8gemm_brq() local
641 for (size_t kr_block_start = (kc & -sr); kr_block_start < kc; in pytorch_pack_swizzle_q8gemm_brq() local
[all …]
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A D | packing.c | 47 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f32_gemm_goi_w() local
94 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f16_gemm_goi_w() local
141 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f32_to_f16_gemm_goi_w() local
198 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_qu8_gemm_goi_w() local
258 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_qs8_gemm_goi_w() local
318 … for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_qs8_gemm_xw_goi_w() local
366 for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f32_gemm_io_w() local
404 for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f16_gemm_io_w() local
442 for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_f32_to_f16_gemm_io_w() local
490 for (size_t kr_block_start = 0; kr_block_start < round_up_po2(kc, skr); kr_block_start += kr) { in xnn_pack_qu8_gemm_io_w() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | fc-unpack.cc | 35 for (size_t kr_block_start = 0; kr_block_start < input_channels_; kr_block_start += kr) { in unpackWeights() local
|