Searched defs:remote_device_mgr (Results 1 – 9 of 9) sorted by relevance
38 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime()64 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
31 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime()52 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
111 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()149 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr()159 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr() function
175 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession() local
260 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
1134 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker()1191 DynamicDeviceMgr* remote_device_mgr, in UpdateRemoteWorker()
118 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr()153 auto remote_device_mgr = absl::make_unique<tensorflow::DynamicDeviceMgr>(); in GetAllRemoteDevices() local162 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr()522 tensorflow::DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateTFE_ContextWithServerDef() local
73 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr() function in tensorflow::__anon7df932810111::TestClusterFLR