• Home
  • Raw
  • Download

Lines Matching refs:nc

18   size_t nc,  in xnn_pack_q8_gemm_goi_w()  argument
30 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_gemm_goi_w()
31 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_goi_w()
62 k += nc * kc; in xnn_pack_q8_gemm_goi_w()
64 b += nc; in xnn_pack_q8_gemm_goi_w()
70 size_t nc, in xnn_pack_q8_gemm_io_w() argument
81 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_gemm_io_w()
82 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_io_w()
102 …const uint8_t kv = k[(kr_block_start + kr_block_offset) * nc + (nr_block_start + nr_block_offset)]; in xnn_pack_q8_gemm_io_w()
117 size_t nc, in xnn_pack_q8_conv_goki_w() argument
130 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_conv_goki_w()
131 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_conv_goki_w()
165 k += ks * kc * nc; in xnn_pack_q8_conv_goki_w()
167 b += nc; in xnn_pack_q8_conv_goki_w()
174 size_t nc, in xnn_pack_q8_conv_kgo_w() argument
186 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_conv_kgo_w()
187 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_conv_kgo_w()
205 k[ki * g * nc + (nr_block_start + nr_block_offset)]; in xnn_pack_q8_conv_kgo_w()
213 k += nc; in xnn_pack_q8_conv_kgo_w()
215 b += nc; in xnn_pack_q8_conv_kgo_w()
222 size_t nc, in xnn_pack_q8_deconv_goki_w() argument
244 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_deconv_goki_w()
245 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_deconv_goki_w()
283 k += kh * kw * kc * nc; in xnn_pack_q8_deconv_goki_w()
285 b += nc; in xnn_pack_q8_deconv_goki_w()
376 size_t nc, in xnn_pack_f16_gemm_goi_w() argument
385 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f16_gemm_goi_w()
386 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_goi_w()
405 k += nc * kc; in xnn_pack_f16_gemm_goi_w()
407 b += nc; in xnn_pack_f16_gemm_goi_w()
413 size_t nc, in xnn_pack_f16_gemm_io_w() argument
421 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f16_gemm_io_w()
422 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_io_w()
434 k[(kr_block_start + kr_block_offset) * nc + (nr_block_start + nr_block_offset)]; in xnn_pack_f16_gemm_io_w()
445 size_t nc, in xnn_pack_f32_gemm_goi_w() argument
458 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemm_goi_w()
459 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_goi_w()
489 k += nc * kc; in xnn_pack_f32_gemm_goi_w()
491 b += nc; in xnn_pack_f32_gemm_goi_w()
497 size_t nc, in xnn_pack_f32_gemm_io_w() argument
509 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemm_io_w()
510 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_io_w()
522 …+ ((kr_block_start + nr_block_offset * kr) & sr_mask) + kr_block_offset) * nc + (nr_block_start + … in xnn_pack_f32_gemm_io_w()
533 k[(kr_block_start + kr_block_offset) * nc + (nr_block_start + nr_block_offset)]; in xnn_pack_f32_gemm_io_w()
544 size_t nc, in xnn_pack_f32_gemminc_goi_w() argument
556 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemminc_goi_w()
557 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemminc_goi_w()
581 k += nc * kc; in xnn_pack_f32_gemminc_goi_w()
587 size_t nc, in xnn_pack_f32_conv_goki_w() argument
601 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_conv_goki_w()
602 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_conv_goki_w()
634 k += ks * kc * nc; in xnn_pack_f32_conv_goki_w()
636 b += nc; in xnn_pack_f32_conv_goki_w()
643 size_t nc, in xnn_pack_f32_conv_kgo_w() argument
652 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_conv_kgo_w()
653 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_conv_kgo_w()
663 k[ki * g * nc + (nr_block_start + nr_block_offset)]; in xnn_pack_f32_conv_kgo_w()
669 k += nc; in xnn_pack_f32_conv_kgo_w()
671 b += nc; in xnn_pack_f32_conv_kgo_w()
677 size_t nc, in xnn_pack_f32_dconv_oki_w() argument
686 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_dconv_oki_w()
687 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_dconv_oki_w()
716 size_t nc, in xnn_pack_f32_deconv_goki_w() argument
739 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_deconv_goki_w()
740 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_deconv_goki_w()
775 k += kh * kw * kc * nc; in xnn_pack_f32_deconv_goki_w()
777 b += nc; in xnn_pack_f32_deconv_goki_w()