Searched refs:ba_stride (Results 1 – 4 of 4) sorted by relevance
110 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_gigemm()136 context->a_offset + batch_index * context->ba_stride, in xnn_compute_igemm()175 context->a_offset + group_index * context->ga_stride + batch_index * context->ba_stride, in xnn_compute_gsubconv2d()213 context->a_offset + batch_index * context->ba_stride, in xnn_compute_subconv2d()
610 ….ba_stride = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_eleme… in setup_conv_path()731 ….ba_stride = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_eleme… in setup_subconv2d_path()
910 ….ba_stride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element… in setup_convolution2d_nhwc()
136 size_t ba_stride; member176 size_t ba_stride; member