Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp624 auto mkldnn_tensor_value = in replaceInputWithMKLDNNTensor() local
628 mkldnn_tensor_value->setDebugName(input->debugName() + "_mkldnn"); in replaceInputWithMKLDNNTensor()
629 n->replaceInputWith(input, mkldnn_tensor_value); in replaceInputWithMKLDNNTensor()
637 auto mkldnn_tensor_value = in replaceInputWithMKLDNNTensor() local
641 mkldnn_tensor_value->setDebugName(input->debugName() + "_mkldnn"); in replaceInputWithMKLDNNTensor()
642 n->replaceInputWith(input, mkldnn_tensor_value); in replaceInputWithMKLDNNTensor()