Lines Matching refs:skr
28 const size_t skr = sr * kr; in xnn_pack_f32_gemm_goi_w() local
29 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f32_gemm_goi_w()
45 …k[(nr_block_start + nr_block_offset) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f32_gemm_goi_w()
82 const size_t skr = sr * kr; in xnn_pack_f16_gemm_goi_w() local
83 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f16_gemm_goi_w()
99 …k[(nr_block_start + nr_block_offset) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f16_gemm_goi_w()
298 const size_t skr = sr * kr; in xnn_pack_f32_gemm_io_w() local
299 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f32_gemm_io_w()
314 …k[(round_down_po2(kr_block_start, skr) + ((kr_block_start + nr_block_offset * kr) & sr_mask) + kr_… in xnn_pack_f32_gemm_io_w()
345 const size_t skr = sr * kr; in xnn_pack_f16_gemm_io_w() local
346 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f16_gemm_io_w()
361 …k[(round_down_po2(kr_block_start, skr) + ((kr_block_start + nr_block_offset * kr) & sr_mask) + kr_… in xnn_pack_f16_gemm_io_w()
442 const size_t skr = sr * kr; in xnn_pack_f32_conv_goki_w() local
443 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f32_conv_goki_w()
460 …_start + nr_block_offset) * ks + ki) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f32_conv_goki_w()
499 const size_t skr = sr * kr; in xnn_pack_f16_conv_goki_w() local
500 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f16_conv_goki_w()
517 …_start + nr_block_offset) * ks + ki) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f16_conv_goki_w()
844 const size_t skr = sr * kr; in xnn_pack_f32_deconv_goki_w() local
845 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f32_deconv_goki_w()
867 …_block_offset) * kh + ky) * kw + kx) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f32_deconv_goki_w()
913 const size_t skr = sr * kr; in xnn_pack_f16_deconv_goki_w() local
914 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f16_deconv_goki_w()
936 …_block_offset) * kh + ky) * kw + kx) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f16_deconv_goki_w()
1355 const size_t skr = sr * kr; in xnn_pack_f32_gemminc_goi_w() local
1356 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f32_gemminc_goi_w()
1366 …k[(nr_block_start + nr_block_offset) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f32_gemminc_goi_w()
1399 const size_t skr = sr * kr; in xnn_pack_f16_gemminc_goi_w() local
1400 const size_t skc = round_down_po2(kc, skr); in xnn_pack_f16_gemminc_goi_w()
1410 …k[(nr_block_start + nr_block_offset) * kc + round_down_po2(kr_block_start, skr) + ((kr_block_start… in xnn_pack_f16_gemminc_goi_w()