Searched defs:remote_device_mgr (Results 1 – 11 of 11) sorted by relevance
39 DeviceMgr* remote_device_mgr) in EagerClusterFunctionLibraryRuntime()79 DeviceMgr* remote_device_mgr() const override { return remote_device_mgr_; } in remote_device_mgr() function
33 DeviceMgr* remote_device_mgr) in ClusterFunctionLibraryRuntime()59 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()148 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr()158 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr() function
218 DeviceMgr* remote_device_mgr = worker_session->remote_device_mgr(); in UpdateSession() local
65 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr()100 auto remote_device_mgr = std::make_unique<tensorflow::DynamicDeviceMgr>(); in GetAllRemoteDevices() local109 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr()487 tensorflow::DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateContextWithServerDef() local
54 auto remote_device_mgr = absl::make_unique<DynamicDeviceMgr>(); in TEST() local
305 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr() function
299 DynamicDeviceMgr remote_device_mgr; in TEST() local
1421 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker()
81 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr() function in tensorflow::__anon96a4e55d0111::TestClusterFLR