Searched refs:nr_block_start (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/src/ |
| D | packing.c | 32 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f32_gemm_goi_w() local
33 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f32_gemm_goi_w()
36 packed_w[nr_block_offset] = b[nr_block_start + nr_block_offset]; 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()
56 k[(nr_block_start + nr_block_offset) * kc + (kr_block_start + kr_block_offset)]; in xnn_pack_f32_gemm_goi_w()
86 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_f16_gemm_goi_w() local
87 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_f16_gemm_goi_w()
90 packed_w[nr_block_offset] = b[nr_block_start + nr_block_offset]; 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()
110 k[(nr_block_start + nr_block_offset) * kc + (kr_block_start + kr_block_offset)]; in xnn_pack_f16_gemm_goi_w()
[all …]
|
| D | operator-run.c | 28 size_t nr_block_start, in xnn_compute_grouped_gemm() argument
42 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_grouped_gemm()
43 …(void*) ((uintptr_t) context->c + mr_block_start * cm_stride + (nr_block_start << context->log2_cs… in xnn_compute_grouped_gemm()
52 size_t nr_block_start, in xnn_compute_gemm() argument
65 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_gemm()
66 …(void*) ((uintptr_t) context->c + mr_block_start * cm_stride + (nr_block_start << context->log2_cs… in xnn_compute_gemm()
95 size_t nr_block_start, in xnn_compute_grouped_batch_igemm() argument
108 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_grouped_batch_igemm()
109 …e + batch_index * context->bc_stride + mr_block_start * cm_stride + (nr_block_start << context->lo… in xnn_compute_grouped_batch_igemm()
121 size_t nr_block_start, in xnn_compute_grouped_igemm() argument
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 90 size_t nr_block_start,
97 size_t nr_block_start,
107 size_t nr_block_start,
115 size_t nr_block_start,
193 size_t nr_block_start,
202 size_t nr_block_start,
209 size_t nr_block_start,
217 size_t nr_block_start,
227 size_t nr_block_start,
237 size_t nr_block_start,
[all …]
|