Searched refs:ba_stride (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 112 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_batch_igemm()
164 context->a_offset + batch_index * context->ba_stride, in xnn_compute_batch_igemm()
223 …ide + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * context->ba_stride), in xnn_compute_grouped_subgemm2d()
260 …->a + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * context->ba_stride), in xnn_compute_subgemm2d()
303 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_subconv2d()
341 context->a_offset + batch_index * context->ba_stride, in xnn_compute_subconv2d()
972 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_hmp_grouped_batch_igemm()
1026 context->a_offset + batch_index * context->ba_stride, in xnn_compute_batch_hmp_igemm()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 176 size_t ba_stride; member
272 size_t ba_stride; member
318 size_t ba_stride; member
|
| /external/XNNPACK/src/operators/ |
| D | deconvolution-nhwc.c | 539 ….ba_stride = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_eleme… in setup_conv_path()
697 .ba_stride = input_height * input_width * input_pixel_stride, in setup_subconv2d_path()
715 .ba_stride = input_height * input_width * input_pixel_stride, in setup_subconv2d_path()
|
| D | convolution-nhwc.c | 987 ….ba_stride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element… in setup_convolution2d_nhwc()
|