Searched refs:workspace_tensors (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1949 std::vector<NodeBuilder::NodeOut>* workspace_tensors,
2217 std::vector<NodeBuilder::NodeOut>* workspace_tensors, in SetUpContiguousInputs() argument
2219 DCHECK(workspace_tensors); in SetUpContiguousInputs()
2304 CHECK_EQ(workspace_tensors->size(), 2); in SetUpContiguousInputs()
2306 nb->Input((*workspace_tensors)[0].node, (*workspace_tensors)[0].index); in SetUpContiguousInputs()
2351 CHECK_EQ(workspace_tensors->size(), 2); in SetUpContiguousInputs()
2353 nb->Input((*workspace_tensors)[1].node, (*workspace_tensors)[1].index); in SetUpContiguousInputs()
2366 std::vector<NodeBuilder::NodeOut> workspace_tensors; in SetUpInputs() local
2399 AddWorkSpaceEdgeIfNeeded(g, old_node, nb, &workspace_tensors, in SetUpInputs()
2412 g, old_node_inputs, nb, old_node, &workspace_tensors, in SetUpInputs()
[all …]
|