Searched refs:ws_tensors (Results 1 – 1 of 1) sorted by relevance
1439 std::vector<NodeBuilder::NodeOut>* ws_tensors,1929 std::vector<NodeBuilder::NodeOut>* ws_tensors, bool* are_ws_tensors_added) { in AddWorkSpaceEdgeIfNeeded() argument1986 CHECK_NOTNULL(ws_tensors); in AddWorkSpaceEdgeIfNeeded()1988 ws_tensors->push_back(NodeBuilder::NodeOut(e->src(), ws.ws_fwd_slot)); in AddWorkSpaceEdgeIfNeeded()1990 ws_tensors->push_back(NodeBuilder::NodeOut( in AddWorkSpaceEdgeIfNeeded()2018 CHECK_NOTNULL(ws_tensors); in AddWorkSpaceEdgeIfNeeded()2020 ws_tensors->push_back(NodeBuilder::NodeOut(dmt_ws, 0)); in AddWorkSpaceEdgeIfNeeded()2022 ws_tensors->push_back(NodeBuilder::NodeOut(dmt_mkl_ws, 0)); in AddWorkSpaceEdgeIfNeeded()