Searched refs:GetNodeProducingMklTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 1398 void GetNodeProducingMklTensor(std::unique_ptr<Graph>* g, Node* orig_node,
1661 GetNodeProducingMklTensor(g, orig_node, n, slot, &mkl_node, in GetNodesProducingMklTensorList()
1674 void MklLayoutRewritePass::GetNodeProducingMklTensor( in GetNodeProducingMklTensor() function in tensorflow::MklLayoutRewritePass
1823 GetNodeProducingMklTensor(g, old_node, conv2d_node, in SetUpContiguousInputs()
1827 GetNodeProducingMklTensor(g, old_node, old_node_inputs[iidx].first, in SetUpContiguousInputs()
|