Home
last modified time | relevance | path

Searched refs:dev_mgr (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc35 TestableDeviceResolverDistributed(const DeviceMgr* dev_mgr, in TestableDeviceResolverDistributed() argument
38 : DeviceResolverDistributed(dev_mgr, worker_cache, task) {} in TestableDeviceResolverDistributed()
64 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker() argument
66 : name_(name), device_mgr_(dev_mgr), device_resolver_(dres) {} in FakeWorker()
160 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
162 new TestableDeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
164 device_mgrs_.push_back(dev_mgr); in DefineWorker()
166 for (auto* d : dev_mgr->ListDevices()) { in DefineWorker()
169 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
Dcollective_param_resolver_distributed_test.cc49 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker() argument
51 : name_(name), device_mgr_(dev_mgr), param_resolver_(cpres) {} in FakeWorker()
165 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
166 device_mgrs_.push_back(dev_mgr); in DefineWorker()
168 for (auto* d : dev_mgr->ListDevices()) { in DefineWorker()
172 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
175 new CollectiveParamResolverDistributed(config, dev_mgr, dev_res, &wc_, in DefineWorker()
178 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, cp_res); in DefineWorker()
Dcollective_rma_distributed_test.cc63 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker() argument
66 device_mgr_(dev_mgr), in FakeWorker()
220 DeviceMgr* dev_mgr = new DeviceMgr(std::move(devices)); in DefineWorker() local
221 device_mgrs_.push_back(dev_mgr); in DefineWorker()
223 for (auto d : dev_mgr->ListDevices()) { in DefineWorker()
227 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
229 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res); in DefineWorker()
Dcollective_rma_distributed.h26 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() argument
30 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
Ddevice_resolver_distributed.cc22 const DeviceMgr* dev_mgr, WorkerCacheInterface* worker_cache, in DeviceResolverDistributed() argument
24 : dev_mgr_(dev_mgr), worker_cache_(worker_cache), task_name_(task_name) {} in DeviceResolverDistributed()
Drpc_collective_executor_mgr.cc29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr() argument
33 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
Ddevice_resolver_distributed.h31 DeviceResolverDistributed(const DeviceMgr* dev_mgr,
Drpc_collective_executor_mgr.h38 const ConfigProto& config, const DeviceMgr* dev_mgr,
Dcollective_param_resolver_distributed.h29 const DeviceMgr* dev_mgr,
Dcollective_param_resolver_distributed.cc82 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverDistributed() argument
85 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc32 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr, in InitializeDeviceAndLocality() argument
35 if (!dev_mgr) { in InitializeDeviceAndLocality()
36 return errors::Internal("Required non-null dev_mgr ", dev_mgr, in InitializeDeviceAndLocality()
40 Status status = dev_mgr->LookupDevice(device_name, device); in InitializeDeviceAndLocality()
46 for (auto d : dev_mgr->ListDevices()) { in InitializeDeviceAndLocality()
Ddevice_resolver_local.h29 DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal() argument
Dcollective_executor_mgr.cc27 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr() argument
30 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
Dcollective_rma_local.h27 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal() argument
30 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
Dbase_collective_executor.h98 int64 step_id, const DeviceMgr* dev_mgr, in BaseCollectiveExecutor() argument
102 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
Dcollective_util.h30 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr,
Dcollective_executor_mgr.h27 CollectiveExecutorMgr(const ConfigProto& config, const DeviceMgr* dev_mgr,
Dhierarchical_tree_broadcaster.cc189 CHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
193 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
Dring_alg.cc244 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
248 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
Dcollective_param_resolver_local.h41 const DeviceMgr* dev_mgr,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.h36 const ConfigProto& config, const DeviceMgr* dev_mgr, in GdrCollectiveExecutorMgr() argument
41 : RpcCollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in GdrCollectiveExecutorMgr()
Dgdr_collective_executor_mgr.cc67 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() argument
72 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc157 const DeviceMgr* dev_mgr, in CollectiveContext() argument
164 dev_mgr(dev_mgr), in CollectiveContext()
Dcollective.h341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
349 const DeviceMgr* dev_mgr; // Not owned variable
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc48 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()

12