Home
last modified time | relevance | path

Searched defs:remote_device_mgr (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.h41 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime()
81 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.h33 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime()
59 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
Dworker_session.cc127 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
164 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr()
174 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
Dworker_session.h44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr() function
Dsession_mgr.cc257 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_distributed_manager.cc75 DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr()
108 auto remote_device_mgr = std::make_unique<DynamicDeviceMgr>(); in GetAllRemoteDevices() local
117 DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr()
483 DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateContextWithServerDef() local
Dexecute_node_test.cc59 auto remote_device_mgr = std::make_unique<DynamicDeviceMgr>(); in TEST() local
Dcontext.h352 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
Dtensor_handle_test.cc299 DynamicDeviceMgr remote_device_mgr; in TEST() local
Dcontext.cc1679 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc83 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr() function in tensorflow::__anon7037703b0111::TestClusterFLR