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
56 const std::unique_ptr<GraphMgr> graph_mgr;
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 | graph_mgr.h | 70 class GraphMgr {
72 explicit GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr);
73 ~GraphMgr();
108 typedef GraphMgr ME;
141 GraphMgr* graph_mgr;
188 TF_DISALLOW_COPY_AND_ASSIGN(GraphMgr);
|
| D | graph_mgr.cc | 56 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr) in GraphMgr() function in tensorflow::GraphMgr
67 GraphMgr::~GraphMgr() { in ~GraphMgr()
71 GraphMgr::Item::~Item() { in ~Item()
101 Status GraphMgr::DecorateAndPublishGraphForDebug( in DecorateAndPublishGraphForDebug()
122 Status GraphMgr::InitItem(const string& session, const GraphDef& gdef, in InitItem()
282 Status GraphMgr::Register(const string& session, const GraphDef& gdef, in Register()
306 Status GraphMgr::Deregister(const string& handle) { in Deregister()
323 Status GraphMgr::DeregisterAll() { in DeregisterAll()
339 Status GraphMgr::SendInputs(const int64 step_id, const NamedTensors& in) { in SendInputs()
355 Status GraphMgr::RecvOutputs(const int64 step_id, NamedTensors* out) { in RecvOutputs()
[all …]
|
| D | worker.cc | 114 GraphMgr::NamedTensors* in, in PrepareRunGraph()
115 GraphMgr::NamedTensors* out) { in PrepareRunGraph()
171 GraphMgr::NamedTensors in; in DoRunGraph()
172 GraphMgr::NamedTensors* out = new GraphMgr::NamedTensors; in DoRunGraph()
283 GraphMgr::NamedTensors in; in DoPartialRunGraph()
284 GraphMgr::NamedTensors* out = new GraphMgr::NamedTensors; in DoPartialRunGraph()
|
| D | session_mgr.cc | 42 std::unique_ptr<GraphMgr>( in SessionMgr()
43 new GraphMgr(worker_env, worker_env->device_mgr)))), in SessionMgr()
88 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession()
97 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession()
|
| D | worker_session.cc | 99 std::unique_ptr<GraphMgr> graph_mgr) in WorkerSession()
113 DeviceMgr* borrowed_device_mgr, std::unique_ptr<GraphMgr> graph_mgr) { in CreateWithBorrowedDeviceMgr()
123 std::unique_ptr<GraphMgr> graph_mgr) in WorkerSession()
|
| D | worker.h | 124 GraphMgr::NamedTensors* in,
125 GraphMgr::NamedTensors* out);
|
| D | cluster_function_library_runtime_test.cc | 45 std::unique_ptr<GraphMgr>())); in ClusterFunctionLibraryRuntimeTest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 74 std::unique_ptr<GraphMgr>()), in RpcRendezvousMgrTest()
|