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