Searched refs:local_device_manager_ (Results 1 – 2 of 2) sorted by relevance
84 local_device_manager_(device_mgr, device_mgr_owned), in EagerContext()1070 if (device_mgr != local_device_manager_.Get()) { in StoreCollectiveOpsServer()1071 if (local_device_manager_.Owned()) { in StoreCollectiveOpsServer()1073 std::move(local_device_manager_.owned_object)); in StoreCollectiveOpsServer()1075 local_device_manager_.Reset(device_mgr); in StoreCollectiveOpsServer()1077 host_cpu_device_ = local_device_manager_.Get()->HostCPU(); in StoreCollectiveOpsServer()1100 local_device_manager_.Get(), env_, /*config=*/config, in StoreCollectiveOpsServer()1315 if (local_device_mgr != local_device_manager_.Get()) { in SetMasterContextState()1316 if (local_device_manager_.Owned()) { in SetMasterContextState()1318 std::move(local_device_manager_.owned_object)); in SetMasterContextState()[all …]
303 return local_device_manager_.Get(); in local_device_mgr()580 OwnedOrUnownedHelper<const DeviceMgr> local_device_manager_; variable