| /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 | collective_rma_local.h | 28 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal() argument
31 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
33 buf_rendezvous_(step_id, dev_mgr), in CollectiveRemoteAccessLocal()
|
| D | device_resolver_local.h | 31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal() argument
|
| D | buf_rendezvous.h | 44 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous() argument
45 : step_id_(step_id), dev_mgr_(dev_mgr) {} in BufRendezvous()
|
| D | collective_executor_mgr.cc | 27 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveExecutorMgr() argument
31 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
|
| D | base_collective_executor.h | 100 const DeviceMgr* dev_mgr, const string* gpu_ring_order, in BaseCollectiveExecutor() argument
104 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
|
| D | permuter.cc | 55 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
59 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|
| D | collective_util.h | 30 Status InitializeDeviceAndLocality(const DeviceMgr* dev_mgr,
|
| D | collective_executor_mgr.h | 30 const ConfigProto& config, const DeviceMgr* dev_mgr,
|
| D | collective_param_resolver_local.h | 44 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 | 246 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
250 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|
| D | permuter_test.cc | 51 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
53 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 65 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker() argument
68 device_mgr_(dev_mgr), in FakeWorker()
70 buf_rendezvous_(kStepId, dev_mgr), in FakeWorker()
248 DeviceMgr* dev_mgr = new StaticDeviceMgr(std::move(devices)); in DefineWorker() local
249 device_mgrs_.push_back(dev_mgr); in DefineWorker()
252 for (auto d : dev_mgr->ListDevices()) { in DefineWorker()
255 DeviceResolverDistributed* dev_res = new DeviceResolverDistributed(dev_mgr); in DefineWorker()
257 FakeWorker* fw = new FakeWorker(worker_name, dev_mgr, dev_res, is_failed); in DefineWorker()
|
| D | collective_rma_distributed.h | 30 const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed() argument
33 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
| D | device_resolver_distributed.cc | 23 DeviceResolverDistributed::DeviceResolverDistributed(const DeviceMgr* dev_mgr) { in DeviceResolverDistributed() argument
25 for (Device* device : dev_mgr->ListDevices()) { in DeviceResolverDistributed()
|
| D | rpc_collective_executor_mgr.cc | 29 const ConfigProto& config, const DeviceMgr* dev_mgr, in RpcCollectiveExecutorMgr() argument
34 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
|
| D | device_resolver_distributed.h | 32 explicit 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 | 32 const DeviceMgr* dev_mgr,
|
| D | collective_param_resolver_distributed_test.cc | 52 FakeWorker(const string& name, DeviceMgr* dev_mgr, in FakeWorker() argument
54 : name_(name), device_mgr_(dev_mgr), param_resolver_(cpres) {} in FakeWorker()
|
| D | collective_param_resolver_distributed.cc | 92 const ConfigProto& config, const DeviceMgr* dev_mgr, in CollectiveParamResolverDistributed() argument
95 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 171 const DeviceMgr* dev_mgr, OpKernelContext* ctx, in CollectiveContext() argument
176 dev_mgr(dev_mgr), in CollectiveContext()
|
| D | collective.h | 367 const DeviceMgr* dev_mgr; // Not owned member
381 const DeviceMgr* dev_mgr, OpKernelContext* ctx,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.cc | 66 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
|