Searched refs:GraphMgr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 29 class GraphMgr; variable
53 GraphMgr* graph_mgr() const { return graph_mgr_.get(); } in graph_mgr()
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,
112 const std::unique_ptr<GraphMgr> graph_mgr_;
|
| D | graph_mgr.h | 70 class GraphMgr {
72 explicit GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr);
73 ~GraphMgr();
109 typedef GraphMgr ME;
142 GraphMgr* graph_mgr;
190 TF_DISALLOW_COPY_AND_ASSIGN(GraphMgr);
|
| D | graph_mgr.cc | 59 GraphMgr::GraphMgr(const WorkerEnv* worker_env, const DeviceMgr* device_mgr) in GraphMgr() function in tensorflow::GraphMgr
70 GraphMgr::~GraphMgr() { in ~GraphMgr()
74 GraphMgr::Item::~Item() { in ~Item()
104 Status GraphMgr::DecorateAndPublishGraphForDebug( in DecorateAndPublishGraphForDebug()
125 Status GraphMgr::InitItem( in InitItem()
295 Status GraphMgr::Register( in Register()
319 Status GraphMgr::Deregister(const string& handle) { in Deregister()
336 Status GraphMgr::DeregisterAll() { in DeregisterAll()
352 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
371 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
[all …]
|
| D | worker.cc | 121 GraphMgr::NamedTensors* in, in PrepareRunGraph()
122 GraphMgr::NamedTensors* out) { in PrepareRunGraph()
184 GraphMgr::NamedTensors in; in DoRunGraph()
185 GraphMgr::NamedTensors* out = new GraphMgr::NamedTensors; in DoRunGraph()
287 GraphMgr::NamedTensors in; in DoPartialRunGraph()
288 GraphMgr::NamedTensors* out = new GraphMgr::NamedTensors; in DoPartialRunGraph()
|
| D | worker_session.cc | 110 std::unique_ptr<DeviceMgr> device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession()
147 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in CreateWithBorrowedDeviceMgr()
157 const DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession()
|
| D | worker.h | 126 GraphMgr::NamedTensors* in,
127 GraphMgr::NamedTensors* out);
|
| D | session_mgr.cc | 43 std::unique_ptr<GraphMgr>( in SessionMgr()
44 new GraphMgr(worker_env, worker_env->device_mgr)), in SessionMgr()
152 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession()
170 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession()
|
| D | cluster_function_library_runtime_test.cc | 48 std::unique_ptr<GraphMgr>(), nullptr)); in ClusterFunctionLibraryRuntimeTest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 127 std::unique_ptr<GraphMgr>(), nullptr), in RpcRendezvousMgrTest()
|