Searched refs:ba_stride (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 384 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_batch_igemm()
436 context->a_offset + batch_index * context->ba_stride, in xnn_compute_batch_igemm()
495 …ide + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * context->ba_stride), in xnn_compute_grouped_subgemm2d()
532 …->a + slice_y * context->ay_stride + slice_x_start * ax_stride + batch_index * context->ba_stride), in xnn_compute_subgemm2d()
575 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_grouped_subconv2d()
613 context->a_offset + batch_index * context->ba_stride, in xnn_compute_subconv2d()
1251 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_hmp_grouped_batch_igemm()
1305 context->a_offset + batch_index * context->ba_stride, in xnn_compute_batch_hmp_igemm()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 282 size_t ba_stride; member
378 size_t ba_stride; member
424 size_t ba_stride; member
|
| /external/XNNPACK/src/operators/ |
| D | deconvolution-nhwc.c | 737 ….ba_stride = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_eleme… in setup_conv_path()
964 .ba_stride = input_height * input_width * input_pixel_stride, in setup_subconv2d_path()
982 .ba_stride = input_height * input_width * input_pixel_stride, in setup_subconv2d_path()
|
| D | convolution-nhwc.c | 1602 ….ba_stride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element… in setup_convolution2d_nhwc()
|