Searched refs:nr_block_size (Results 1 – 3 of 3) sorted by relevance
33 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_goi_w() local35 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_f32_gemm_goi_w()42 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_f32_gemm_goi_w()48 packed_w += (nr - nr_block_size) * kr; in xnn_pack_f32_gemm_goi_w()53 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_f32_gemm_goi_w()60 packed_w += (nr - nr_block_size) * kr; in xnn_pack_f32_gemm_goi_w()87 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_goi_w() local89 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_f16_gemm_goi_w()96 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_f16_gemm_goi_w()102 packed_w += (nr - nr_block_size) * kr; in xnn_pack_f16_gemm_goi_w()[all …]
30 size_t nr_block_size) in xnn_compute_grouped_gemm() argument38 nr_block_size, in xnn_compute_grouped_gemm()54 size_t nr_block_size) in xnn_compute_gemm() argument61 nr_block_size, in xnn_compute_gemm()97 size_t nr_block_size) in xnn_compute_grouped_batch_igemm() argument104 nr_block_size, in xnn_compute_grouped_batch_igemm()123 size_t nr_block_size) in xnn_compute_grouped_igemm() argument130 nr_block_size, in xnn_compute_grouped_igemm()149 size_t nr_block_size) in xnn_compute_batch_igemm() argument156 nr_block_size, in xnn_compute_batch_igemm()[all …]
92 size_t nr_block_size);99 size_t nr_block_size);109 size_t nr_block_size);117 size_t nr_block_size);195 size_t nr_block_size);204 size_t nr_block_size);211 size_t nr_block_size);219 size_t nr_block_size);229 size_t nr_block_size);239 size_t nr_block_size);[all …]