Searched refs:input_stride_h (Results 1 – 4 of 4) sorted by relevance
67 const int input_stride_h = src->info()->strides_in_bytes().z() / element_size; in convolve_nhwc() local152 … + id[3] * input_stride_n + in_w_start * input_stride_w + in_h_start * input_stride_h; in convolve_nhwc()157 …x_h = index_h_start; index_h < index_h_end; ++index_h, in_ptr_row += input_stride_h, weights_ptr_r… in convolve_nhwc()217 const T *const in_ptr_row = in_ptr_start + index_in_h * input_stride_h; in convolve_nhwc()
74 const int input_stride_h = src->info()->strides_in_bytes()[1] / element_size; in convolve_nchw() local143 const T *in_ptr_row = in_ptr_row_0 + index_in_h * input_stride_h; in convolve_nchw()
53 const int input_stride_h = src->info()->strides_in_bytes().z() / element_size; in directconv3d_float_neon_ndhwc() local139 const T *const in_ptr_row = in_ptr_d + index_in_h * input_stride_h; in directconv3d_float_neon_ndhwc()
69 const int input_stride_h = src->info()->strides_in_bytes().z() / element_size; in directconv3d_quantized_neon_ndhwc() local155 const T *const in_ptr_row = in_ptr_d + index_in_h * input_stride_h; in directconv3d_quantized_neon_ndhwc()