Lines Matching refs:group_index

26     size_t group_index,  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()
42 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->wg_s… in xnn_compute_grouped_gemm()
43 …mr_block_start * cm_stride + (nr_block_start << context->log2_csize) + group_index * context->cg_s… in xnn_compute_grouped_gemm()
93 size_t group_index, in xnn_compute_grouped_batch_igemm() argument
108 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_grouped_batch_igemm()
109 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + batch_index * context->bc_str… in xnn_compute_grouped_batch_igemm()
112 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_batch_igemm()
119 size_t group_index, in xnn_compute_grouped_igemm() argument
134 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_grouped_igemm()
135 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + mr_block_start * cm_stride + … in xnn_compute_grouped_igemm()
138 context->a_offset + group_index * context->ga_stride, in xnn_compute_grouped_igemm()
197 size_t group_index, in xnn_compute_grouped_subgemm2d() argument
223 …(const void*) ((uintptr_t) context->a + group_index * context->ga_stride + slice_y * context->ay_s… in xnn_compute_grouped_subgemm2d()
225 …n_params->weights + nc_block_start * subconvolution_params->w_stride + group_index * context->gw_s… in xnn_compute_grouped_subgemm2d()
226 …(void*) ((uintptr_t) subconvolution_params->output + group_index * context->gc_stride + slice_y * … in xnn_compute_grouped_subgemm2d()
272 size_t group_index, in xnn_compute_grouped_subconv2d() argument
299 …n_params->weights + nc_block_start * subconvolution_params->w_stride + group_index * context->gw_s… in xnn_compute_grouped_subconv2d()
300 …(void*) ((uintptr_t) subconvolution_params->output + group_index * context->gc_stride + slice_y * … in xnn_compute_grouped_subconv2d()
303 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_subconv2d()
902 size_t group_index, in xnn_compute_hmp_grouped_gemm() argument
916 (const void*) ((uintptr_t) context->a + mr_block_start * a_stride + group_index * k_scaled), in xnn_compute_hmp_grouped_gemm()
918 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->wg_s… in xnn_compute_hmp_grouped_gemm()
919 …mr_block_start * cm_stride + (nr_block_start << context->log2_csize) + group_index * context->cg_s… in xnn_compute_hmp_grouped_gemm()
953 size_t group_index, in xnn_compute_hmp_grouped_batch_igemm() argument
968 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_hmp_grouped_batch_igemm()
969 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + batch_index * context->bc_str… in xnn_compute_hmp_grouped_batch_igemm()
972 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_hmp_grouped_batch_igemm()
980 size_t group_index, in xnn_compute_hmp_grouped_igemm() argument
995 … ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * context->gw_s… in xnn_compute_hmp_grouped_igemm()
996 …(void*) ((uintptr_t) context->c + group_index * context->gc_stride + mr_block_start * cm_stride + … in xnn_compute_hmp_grouped_igemm()
999 context->a_offset + group_index * context->ga_stride, in xnn_compute_hmp_grouped_igemm()