Searched refs:GetNodeProducingMklTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1940 void GetNodeProducingMklTensor(std::unique_ptr<Graph>* g,
2180 GetNodeProducingMklTensor(g, orig_node, n, slot, &mkl_node, in GetNodesProducingMklTensorList()
2193 void MklLayoutRewritePass::GetNodeProducingMklTensor( in GetNodeProducingMklTensor() function in tensorflow::MklLayoutRewritePass
2341 GetNodeProducingMklTensor(g, old_node, conv2d_node, in SetUpContiguousInputs()
2345 GetNodeProducingMklTensor(g, old_node, old_node_inputs[iidx].first, in SetUpContiguousInputs()
|