Searched refs:nr_block_start (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/src/xnnpack/ |
| D | pack.h | 30 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_gemm_goi_w() local
31 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_goi_w()
35 *((int32_t*) packed_w) = b[nr_block_start + nr_block_offset] + boff; in xnn_pack_q8_gemm_goi_w()
51 …const uint8_t kv = k[(nr_block_start + nr_block_offset) * kc + (kr_block_start + kr_block_offset)]; in xnn_pack_q8_gemm_goi_w()
81 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_gemm_io_w() local
82 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_io_w()
86 *((int32_t*) packed_w) = b[nr_block_start + nr_block_offset] + boff; 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()
130 for (size_t nr_block_start = 0; nr_block_start < nc; nr_block_start += nr) { in xnn_pack_q8_conv_goki_w() local
131 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_conv_goki_w()
[all …]
|
| D | compute.h | 72 size_t nr_block_start,
79 size_t nr_block_start,
152 size_t nr_block_start,
160 size_t nr_block_start,
194 size_t nr_block_start,
204 size_t nr_block_start,
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 27 size_t nr_block_start, in xnn_compute_ggemm() argument
41 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_ggemm()
42 …(void*) ((uintptr_t) context->c + mr_block_start * cm_stride + (nr_block_start << context->log2_cs… in xnn_compute_ggemm()
51 size_t nr_block_start, in xnn_compute_gemm() argument
64 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_gemm()
65 …(void*) ((uintptr_t) context->c + mr_block_start * cm_stride + (nr_block_start << context->log2_cs… in xnn_compute_gemm()
93 size_t nr_block_start, in xnn_compute_gigemm() argument
106 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_gigemm()
107 …e + batch_index * context->bc_stride + mr_block_start * cm_stride + (nr_block_start << context->lo… in xnn_compute_gigemm()
119 size_t nr_block_start, in xnn_compute_igemm() argument
[all …]
|