Searched refs:workspace_tensors (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 1417 std::vector<NodeBuilder::NodeOut>* workspace_tensors,
1707 std::vector<NodeBuilder::NodeOut>* workspace_tensors, in SetUpContiguousInputs() argument
1709 CHECK_NOTNULL(workspace_tensors); in SetUpContiguousInputs()
1796 CHECK_EQ(workspace_tensors->size(), 2); in SetUpContiguousInputs()
1798 nb->Input((*workspace_tensors)[0].node, (*workspace_tensors)[0].index); in SetUpContiguousInputs()
1842 CHECK_EQ(workspace_tensors->size(), 2); in SetUpContiguousInputs()
1844 nb->Input((*workspace_tensors)[1].node, (*workspace_tensors)[1].index); in SetUpContiguousInputs()
1857 std::vector<NodeBuilder::NodeOut> workspace_tensors; in SetUpInputs() local
1878 AddWorkSpaceEdgeIfNeeded(g, old_node, nb, &workspace_tensors, in SetUpInputs()
1891 g, old_node_inputs, nb, old_node, &workspace_tensors, in SetUpInputs()
|