Home
last modified time | relevance | path

Searched defs:mkldnn_input (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/
DLinear.cpp219 const ideep::tensor mkldnn_input = itensor_view_from_dense(input_reshaped); in mkldnn_linear_pointwise() local
307 const ideep::tensor mkldnn_input = itensor_view_from_dense(input_reshaped); in mkldnn_linear_pointwise_binary() local
DConvPrepack.cpp162 const ideep::tensor mkldnn_input = itensor_from_tensor(input); in mkldnn_convolution_out() local