Searched refs:mr_block_start (Results 1 – 2 of 2) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 26 size_t mr_block_start, in xnn_compute_ggemm() argument
39 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), 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()
50 size_t mr_block_start, in xnn_compute_gemm() argument
62 (const void*) ((uintptr_t) context->a + mr_block_start * a_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()
74 size_t mr_block_start, in xnn_compute_spmm() argument
80 … ((uintptr_t) context->a + batch_index * context->batched_a_stride + mr_block_start * sizeof(float… in xnn_compute_spmm()
84 …(void*) ((uintptr_t) context->c + batch_index * context->batched_c_stride + mr_block_start * sizeo… in xnn_compute_spmm()
92 size_t mr_block_start, in xnn_compute_gigemm() argument
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 71 size_t mr_block_start,
78 size_t mr_block_start,
117 size_t mr_block_start,
151 size_t mr_block_start,
159 size_t mr_block_start,
|