Searched refs:GetNodeProducingMklTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | mkl_layout_pass.cc | 1891 void GetNodeProducingMklTensor(std::unique_ptr<Graph>* g,
2131 GetNodeProducingMklTensor(g, orig_node, n, slot, &mkl_node, in GetNodesProducingMklTensorList()
2144 void MklLayoutRewritePass::GetNodeProducingMklTensor( in GetNodeProducingMklTensor() function in tensorflow::MklLayoutRewritePass
2292 GetNodeProducingMklTensor(g, old_node, conv2d_node, in SetUpContiguousInputs()
2296 GetNodeProducingMklTensor(g, old_node, old_node_inputs[iidx].first, in SetUpContiguousInputs()
|