Searched refs:mkl_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 1399 Node* n, int n_output_slot, Node** mkl_node,
1659 Node* mkl_node = nullptr; in GetNodesProducingMklTensorList() local
1661 GetNodeProducingMklTensor(g, orig_node, n, slot, &mkl_node, in GetNodesProducingMklTensorList()
1664 NodeBuilder::NodeOut(mkl_node, mkl_node_output_slot)); in GetNodesProducingMklTensorList()
1676 Node** mkl_node, int* mkl_node_output_slot) { in GetNodeProducingMklTensor() argument
1678 CHECK_NOTNULL(mkl_node); in GetNodeProducingMklTensor()
1689 *mkl_node = n; in GetNodeProducingMklTensor()
1697 GetDummyMklTensorNode(g, mkl_node, orig_node); in GetNodeProducingMklTensor()
1698 CHECK_NOTNULL(*mkl_node); in GetNodeProducingMklTensor()
1818 Node* mkl_node = nullptr; in SetUpContiguousInputs() local
[all …]
|