Searched refs:graph_mgr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 53 GraphMgr* graph_mgr() const { return graph_mgr_.get(); } in graph_mgr() function
62 std::unique_ptr<GraphMgr> graph_mgr,
68 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr,
94 std::unique_ptr<GraphMgr> graph_mgr,
|
| D | worker_session.cc | 110 std::unique_ptr<DeviceMgr> device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession() argument
115 graph_mgr_(std::move(graph_mgr)), in WorkerSession()
147 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in CreateWithBorrowedDeviceMgr() argument
151 std::move(graph_mgr), std::move(remote_device_mgr))); in CreateWithBorrowedDeviceMgr()
157 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession() argument
162 graph_mgr_(std::move(graph_mgr)), in WorkerSession()
|
| D | worker.cc | 81 s = session->graph_mgr()->Register( in RegisterGraphAsync()
102 s = session->graph_mgr()->Deregister(request->graph_handle()); in DeregisterGraphAsync()
224 session->graph_mgr()->ExecuteAsync( in DoRunGraph()
231 s = session->graph_mgr()->RecvOutputs(step_id, out); in DoRunGraph()
317 session->graph_mgr()->ExecuteAsync( in DoPartialRunGraph()
326 s = session->graph_mgr()->SendInputs(step_id, in); in DoPartialRunGraph()
333 session->graph_mgr()->RecvOutputsAsync( in DoPartialRunGraph()
|
| D | session_mgr.cc | 152 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession() local
156 std::move(device_mgr), std::move(graph_mgr), in CreateSession()
170 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession() local
174 worker_env_->device_mgr, std::move(graph_mgr), in CreateSession()
|
| D | BUILD | 99 ":graph_mgr",
139 ":graph_mgr",
209 ":graph_mgr",
429 name = "graph_mgr",
430 srcs = ["graph_mgr.cc"],
431 hdrs = ["graph_mgr.h"],
|
| D | graph_mgr.h | 142 GraphMgr* graph_mgr; member
|
| D | graph_mgr.cc | 77 if (!graph_mgr->skip_cost_models_) { in ~Item()
78 graph_mgr->cost_model_manager_.RemoveCostModelForGraph(unit.graph.get()); in ~Item()
211 item->graph_mgr = this; in InitItem()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 220 "//tensorflow/core/distributed_runtime:graph_mgr",
330 "//tensorflow/core/distributed_runtime:graph_mgr",
|