Searched refs:ax_stride (Results 1 – 3 of 3) sorted by relevance
217 const size_t ax_stride = context->ax_stride; in xnn_compute_grouped_subgemm2d() local223 …x * context->ga_stride + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * … in xnn_compute_grouped_subgemm2d()224 ax_stride, in xnn_compute_grouped_subgemm2d()254 const size_t ax_stride = context->ax_stride; in xnn_compute_subgemm2d() local260 …(uintptr_t) context->a + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * … in xnn_compute_subgemm2d()261 ax_stride, in xnn_compute_subgemm2d()
264 size_t ax_stride; member
689 .ax_stride = input_pixel_stride, in setup_subconv2d_path()