Searched refs:mkl_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1931 int n_output_slot, Node** mkl_node,
2169 Node* mkl_node = nullptr; in GetNodesProducingMklTensorList() local
2171 GetNodeProducingMklTensor(g, orig_node, n, slot, &mkl_node, in GetNodesProducingMklTensorList()
2174 NodeBuilder::NodeOut(mkl_node, mkl_node_output_slot)); in GetNodesProducingMklTensorList()
2186 int n_output_slot, Node** mkl_node, int* mkl_node_output_slot) { in GetNodeProducingMklTensor() argument
2188 DCHECK(mkl_node); in GetNodeProducingMklTensor()
2199 *mkl_node = n; in GetNodeProducingMklTensor()
2207 GetDummyMklTensorNode(g, mkl_node, orig_node); in GetNodeProducingMklTensor()
2208 DCHECK(*mkl_node); in GetNodeProducingMklTensor()
2327 Node* mkl_node = nullptr; in SetUpContiguousInputs() local
[all …]
|