Searched refs:replaceInputWithMKLDNNTensor (Results 1 – 1 of 1) sorted by relevance
621 void replaceInputWithMKLDNNTensor(Node* n, size_t index) { in replaceInputWithMKLDNNTensor() function632 void replaceInputWithMKLDNNTensor( in replaceInputWithMKLDNNTensor() function645 void replaceInputWithMKLDNNTensor(Node* n, const std::string& name) { in replaceInputWithMKLDNNTensor() function648 replaceInputWithMKLDNNTensor(n, name, mkldnn_tensor); in replaceInputWithMKLDNNTensor()671 replaceInputWithMKLDNNTensor(conv, "weight", conv_w_mkldnn); in moveConvWeightsToMKLDNN()674 replaceInputWithMKLDNNTensor(conv, "bias"); in moveConvWeightsToMKLDNN()689 replaceInputWithMKLDNNTensor(n, i); in moveWeightsToMKLDNN()