Searched refs:local_device_mgr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 85 new DeviceResolverLocal(local_device_mgr())); in EagerContext()
87 opts.config, local_device_mgr(), drl.get(), in EagerContext()
90 opts.config, local_device_mgr(), std::move(drl), std::move(cprl))); in EagerContext()
464 DeviceMgr* local_device_mgr, int keep_alive_secs) { in InitializeRemote() argument
475 local_unowned_device_manager_ = local_device_mgr; in InitializeRemote()
|
| D | context.h | 172 const tensorflow::DeviceMgr* local_device_mgr() const { in local_device_mgr() function
223 DeviceMgr* local_device_mgr, int keep_alive_secs);
|
| D | execute.cc | 248 return ctx->local_device_mgr()->LookupDevice(d->name(), &tmp).ok(); in IsLocal()
282 auto status = ctx->local_device_mgr()->LookupDevice(device_name, device); in FindDeviceFromName()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 369 ctx->context.local_device_mgr()->ListDeviceAttributes(&list->response); in TFE_ContextListDevices()
|