Home
last modified time | relevance | path

Searched refs:remote_device_mgr_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.h42 remote_device_mgr_(remote_device_mgr) {} in EagerClusterFunctionLibraryRuntime()
79 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr()
84 DeviceMgr* remote_device_mgr_; // not owned. variable
Deager_service_impl_test.cc706 remote_device_mgr_ = absl::make_unique<StaticDeviceMgr>( in FunctionWithRemoteInputsTest()
780 remote_device_mgr_.get(), Env::Default(), /*config=*/ in Init()
826 std::unique_ptr<DeviceMgr> remote_device_mgr_; member in tensorflow::eager::__anon09d935ad0111::FunctionWithRemoteInputsTest
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.h36 remote_device_mgr_(remote_device_mgr) {} in ClusterFunctionLibraryRuntime()
59 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr()
73 DeviceMgr* remote_device_mgr_; // not owned. variable
Dworker_session.cc121 remote_device_mgr_(std::move(remote_device_mgr)) { in WorkerSession()
137 TF_RETURN_IF_ERROR(remote_device_mgr_->RemoveDevices(removed_remote_devices)); in UpdateWorkerCacheAndDevices()
139 remote_device_mgr_->AddDevices(std::move(added_remote_devices))); in UpdateWorkerCacheAndDevices()
167 remote_device_mgr_(std::move(remote_device_mgr)) { in WorkerSession()
Dworker_session.h44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr()
118 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr_; variable