/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | device_resolver_distributed_test.cc | 35 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()
|
D | collective_param_resolver_distributed_test.cc | 49 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()
|
D | collective_rma_distributed_test.cc | 63 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()
|
D | collective_rma_distributed.h | 26 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() argument 30 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
D | device_resolver_distributed.cc | 22 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()
|
D | rpc_collective_executor_mgr.cc | 29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr() argument 33 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
|
D | device_resolver_distributed.h | 31 DeviceResolverDistributed(const DeviceMgr* dev_mgr,
|
D | rpc_collective_executor_mgr.h | 38 const ConfigProto& config, const DeviceMgr* dev_mgr,
|
D | collective_param_resolver_distributed.h | 29 const DeviceMgr* dev_mgr,
|
D | collective_param_resolver_distributed.cc | 82 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/ |
D | collective_util.cc | 32 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()
|
D | device_resolver_local.h | 29 DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal() argument
|
D | collective_executor_mgr.cc | 27 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr() argument 30 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
|
D | collective_rma_local.h | 27 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal() argument 30 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
|
D | base_collective_executor.h | 98 int64 step_id, const DeviceMgr* dev_mgr, in BaseCollectiveExecutor() argument 102 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
|
D | collective_util.h | 30 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr,
|
D | collective_executor_mgr.h | 27 CollectiveExecutorMgr(const ConfigProto& config, const DeviceMgr* dev_mgr,
|
D | hierarchical_tree_broadcaster.cc | 189 CHECK(col_ctx->dev_mgr); in InitializeCollectiveContext() 193 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|
D | ring_alg.cc | 244 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext() 248 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|
D | collective_param_resolver_local.h | 41 const DeviceMgr* dev_mgr,
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_collective_executor_mgr.h | 36 const ConfigProto& config, const DeviceMgr* dev_mgr, in GdrCollectiveExecutorMgr() argument 41 : RpcCollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in GdrCollectiveExecutorMgr()
|
D | gdr_collective_executor_mgr.cc | 67 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() argument 72 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
/external/tensorflow/tensorflow/core/framework/ |
D | collective.cc | 157 const DeviceMgr* dev_mgr, in CollectiveContext() argument 164 dev_mgr(dev_mgr), in CollectiveContext()
|
D | collective.h | 341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr, 349 const DeviceMgr* dev_mgr; // Not owned variable
|
/external/tensorflow/tensorflow/core/kernels/ |
D | collective_nccl_reducer.cc | 48 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|