Searched refs:kr_block_start (Results 1 – 1 of 1) sorted by relevance
| /external/XNNPACK/src/ |
| D | packing.c | 45 … 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
48 …const size_t kc_idx = round_down_po2(kr_block_start, skr) + ((kr_block_start + kr_block_offset + n… in xnn_pack_f32_gemm_goi_w()
92 … 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
95 …const size_t kc_idx = round_down_po2(kr_block_start, skr) + ((kr_block_start + kr_block_offset + n… in xnn_pack_f16_gemm_goi_w()
139 … 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
142 …const size_t kc_idx = round_down_po2(kr_block_start, skr) + ((kr_block_start + kr_block_offset + n… in xnn_pack_f32_to_f16_gemm_goi_w()
196 … 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
200 …const size_t kc_idx = round_down_po2(kr_block_start, skr) + ((kr_block_start + kr_block_offset + n… in xnn_pack_qu8_gemm_goi_w()
256 … 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
260 …const size_t kc_idx = round_down_po2(kr_block_start, skr) + ((kr_block_start + kr_block_offset + n… in xnn_pack_qs8_gemm_goi_w()
[all …]
|