Searched refs:graph_mgr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.cc | 99 std::unique_ptr<GraphMgr> graph_mgr) in WorkerSession() argument
103 graph_mgr(std::move(graph_mgr)), in WorkerSession()
113 DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr) { in CreateWithBorrowedDeviceMgr() argument
116 borrowed_device_mgr, std::move(graph_mgr))); in CreateWithBorrowedDeviceMgr()
123 std::unique_ptr<GraphMgr> graph_mgr) in WorkerSession() argument
127 graph_mgr(std::move(graph_mgr)), in WorkerSession()
134 if (graph_mgr) { in ~WorkerSession()
135 Status s = graph_mgr->DeregisterAll(); in ~WorkerSession()
|
| D | worker_session.h | 56 const std::unique_ptr<GraphMgr> graph_mgr; member
63 std::unique_ptr<GraphMgr> graph_mgr);
68 DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr);
76 std::unique_ptr<GraphMgr> graph_mgr);
|
| D | session_mgr.cc | 88 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession() local
92 std::move(device_mgr), std::move(graph_mgr))); in CreateSession()
97 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession() local
101 worker_env_->device_mgr, std::move(graph_mgr)); in CreateSession()
|
| D | worker.cc | 74 s = session->graph_mgr->Register( in RegisterGraphAsync()
95 s = session->graph_mgr->Deregister(request->graph_handle()); in DeregisterGraphAsync()
225 session->graph_mgr->ExecuteAsync( in DoRunGraph()
231 s = session->graph_mgr->RecvOutputs(step_id, out); in DoRunGraph()
312 session->graph_mgr->ExecuteAsync( in DoPartialRunGraph()
321 s = session->graph_mgr->SendInputs(step_id, in); in DoPartialRunGraph()
328 session->graph_mgr->RecvOutputsAsync( in DoPartialRunGraph()
|
| D | BUILD | 96 ":graph_mgr",
129 ":graph_mgr",
198 ":graph_mgr",
417 name = "graph_mgr",
418 srcs = ["graph_mgr.cc"],
419 hdrs = ["graph_mgr.h"],
|
| D | graph_mgr.cc | 74 if (!graph_mgr->skip_cost_models_) { in ~Item()
75 graph_mgr->cost_model_manager_.RemoveCostModelForGraph(unit.graph); in ~Item()
199 item->graph_mgr = this; in InitItem()
|
| D | graph_mgr.h | 141 GraphMgr* graph_mgr; member
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | BUILD | 66 "//tensorflow/core/distributed_runtime:graph_mgr",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 191 "//tensorflow/core/distributed_runtime:graph_mgr",
296 "//tensorflow/core/distributed_runtime:graph_mgr",
|