Searched refs:mr_block_start (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 27 size_t mr_block_start, in xnn_compute_grouped_gemm() argument
40 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), 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()
51 size_t mr_block_start, in xnn_compute_gemm() argument
63 (const void*) ((uintptr_t) context->a + mr_block_start * a_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()
75 size_t mr_block_start, in xnn_compute_spmm() argument
81 …void*) ((uintptr_t) context->input + batch_index * context->batched_input_stride + mr_block_start), in xnn_compute_spmm()
85 …id*) ((uintptr_t) context->output + batch_index * context->batched_output_stride + mr_block_start), in xnn_compute_spmm()
94 size_t mr_block_start, in xnn_compute_grouped_batch_igemm() argument
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 89 size_t mr_block_start,
96 size_t mr_block_start,
106 size_t mr_block_start,
114 size_t mr_block_start,
157 size_t mr_block_start,
192 size_t mr_block_start,
201 size_t mr_block_start,
208 size_t mr_block_start,
216 size_t mr_block_start,
226 size_t mr_block_start,
[all …]
|