Searched refs:kernel_stride_h (Results 1 – 4 of 4) sorted by relevance
75 const unsigned int kernel_stride_h = weights->info()->strides_in_bytes().z() / element_size; in convolve_nhwc() local153 …onst T *weights_ptr_row = reinterpret_cast<const T *>(wei.ptr()) + index_h_start * kernel_stride_h; in convolve_nhwc()157 …index_h < index_h_end; ++index_h, in_ptr_row += input_stride_h, weights_ptr_row += kernel_stride_h) in convolve_nhwc()218 … const T *const weights_ptr_row = weights_ptr_start + index_wei_h * kernel_stride_h; in convolve_nhwc()
84 const unsigned int kernel_stride_h = weights->info()->strides_in_bytes().y() / element_size; in convolve_nchw() local144 const T *weights_ptr_row = weights_ptr_row_0 + index_wei_h * kernel_stride_h; in convolve_nchw()
62 const unsigned int kernel_stride_h = weights->info()->strides_in_bytes()[3] / element_size; in directconv3d_float_neon_ndhwc() local140 const T *const weights_ptr_row = weights_ptr_d + index_wei_h * kernel_stride_h; in directconv3d_float_neon_ndhwc()
78 const unsigned int kernel_stride_h = weights->info()->strides_in_bytes()[3] / element_size; in directconv3d_quantized_neon_ndhwc() local156 const T *const weights_ptr_row = weights_ptr_d + index_wei_h * kernel_stride_h; in directconv3d_quantized_neon_ndhwc()