Searched refs:local_device_mgr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 116 new DeviceResolverLocal(local_device_mgr())); in EagerContext()
118 opts.config, local_device_mgr(), drl.get(), in EagerContext()
121 new CollectiveExecutorMgr(opts.config, local_device_mgr(), std::move(drl), in EagerContext()
146 for (Device* d : local_device_mgr()->ListDevices()) { in InitPrioritizedDeviceTypeList()
474 local_device_mgr()->ListDeviceAttributes(devices); in ListDevices()
486 auto local_devices = local_device_mgr()->ListDevices(); in StartStep()
683 auto status = local_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
919 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs, in InitializeRemoteMaster() argument
937 context_id, 0, r, local_device_mgr, keep_alive_secs, cluster_flr, in InitializeRemoteMaster()
946 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs, in UpdateRemoteMaster() argument
[all …]
|
| D | context.h | 242 return new IntraProcessRendezvous(local_device_mgr()); in CreateRendezvous()
257 const tensorflow::DeviceMgr* local_device_mgr() const { in local_device_mgr() function
321 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs,
337 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs,
561 DeviceMgr* local_device_mgr, int keep_alive_secs,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.cc | 147 if (parent_->local_device_mgr()->LookupDevice(in.op_device(), &device).ok() || in DeserializeRemoteTensorHandle()
148 parent_->local_device_mgr()->LookupDevice(in.device(), &device).ok()) { in DeserializeRemoteTensorHandle()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 174 DeviceMgr* local_device_mgr = worker_session->device_mgr(); in UpdateSession() local
183 if (!local_device_mgr->LookupDevice(da.name(), &device).ok() && in UpdateSession()
|