| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| 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 | rpc_collective_executor_mgr.cc | 30 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in RpcCollectiveExecutorMgr() argument
34 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
|
| D | rpc_collective_executor_mgr.h | 39 std::unique_ptr<DeviceResolverDistributed> dev_resolver,
|
| D | collective_param_resolver_distributed.h | 33 DeviceResolverDistributed* dev_resolver,
|
| D | collective_rma_distributed_test.cc | 233 DeviceResolverDistributed* dev_resolver = dev_resolver_item.second; in ResolveDeviceAttributes() local
235 TF_CHECK_OK(dev_resolver->UpdateDeviceAttributes(item.second)); in ResolveDeviceAttributes()
|
| D | collective_param_resolver_distributed.cc | 93 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed() argument
95 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.h | 29 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessLocal() argument
32 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()
|
| D | collective_executor_mgr.cc | 28 std::unique_ptr<DeviceResolverInterface> dev_resolver, in CollectiveExecutorMgr() argument
32 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()
|
| D | collective_executor_mgr.h | 31 std::unique_ptr<DeviceResolverInterface> dev_resolver,
|
| D | collective_param_resolver_local.h | 45 DeviceResolverInterface* dev_resolver,
|
| 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()
|
| D | ring_gatherer_test.cc | 47 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
49 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| D | ring_reducer_test.cc | 48 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
50 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| D | collective_param_resolver_local.cc | 43 DeviceResolverInterface* dev_resolver, const string& task_name) in CollectiveParamResolverLocal() argument
46 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()
|
| D | hierarchical_tree_broadcaster_test.cc | 141 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
143 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 282 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init() local
286 dev_resolver.get(), worker_cache, in Init()
289 config, worker_env_.device_mgr, std::move(dev_resolver), in Init()
451 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in UpdateServerDef() local
456 dev_resolver.get(), worker_cache, default_worker_name)); in UpdateServerDef()
459 std::move(dev_resolver), std::move(param_resolver), in UpdateServerDef()
|