Home
last modified time | relevance | path

Searched refs:may_convert_to_default_contiguous_strides (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/
DLinear.cpp194 input = may_convert_to_default_contiguous_strides(input); in mkldnn_linear_pointwise()
272 input = may_convert_to_default_contiguous_strides(input); in mkldnn_linear_pointwise_binary()
291 other_reshaped = may_convert_to_default_contiguous_strides(other_reshaped); in mkldnn_linear_pointwise_binary()
DUtils.h63 inline Tensor may_convert_to_default_contiguous_strides(const Tensor& input) { in may_convert_to_default_contiguous_strides() function
DMatmul.cpp302 mat1_ = may_convert_to_default_contiguous_strides(mat1_); in mkldnn_matmul()
303 mat2_ = may_convert_to_default_contiguous_strides(mat2_); in mkldnn_matmul()