Searched defs:mkldnn_output (Results 1 – 3 of 3) sorted by relevance
148 ideep::tensor& mkldnn_output, in mkldnn_convolution_out()206 ideep::tensor mkldnn_output = itensor_from_tensor(output); in run() local247 ideep::tensor mkldnn_output = {o_desc, output}; in run() local
213 ideep::tensor mkldnn_output = itensor_from_tensor(output); in mkldnn_linear_pointwise() local305 ideep::tensor mkldnn_output = itensor_from_tensor(output); in mkldnn_linear_pointwise_binary() local
542 auto mkldnn_output = mkldnn_rnn( in mkldnn_impl() local