| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.h | 27 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed() argument
30 : 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
33 : 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 | 30 DeviceResolverDistributed* dev_resolver,
|
| D | collective_param_resolver_distributed.cc | 83 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed() argument
85 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.h | 37 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in GdrCollectiveExecutorMgr() argument
41 : RpcCollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in GdrCollectiveExecutorMgr()
|
| D | gdr_server_lib.cc | 71 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init() local
76 config, env->device_mgr, dev_resolver.get(), worker_cache, in Init()
79 config, env->device_mgr, std::move(dev_resolver), in Init()
|
| D | gdr_collective_executor_mgr.cc | 68 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessDistributed() argument
72 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 28 std::unique_ptr<DeviceResolverInterface> dev_resolver, in CollectiveExecutorMgr() argument
31 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()
|
| D | collective_rma_local.h | 28 DeviceResolverInterface* dev_resolver, in CollectiveRemoteAccessLocal() argument
31 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()
|
| D | collective_executor_mgr.h | 28 std::unique_ptr<DeviceResolverInterface> dev_resolver,
|
| D | collective_param_resolver_local.h | 42 DeviceResolverInterface* dev_resolver,
|
| D | ring_gatherer_test.cc | 46 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
48 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| D | collective_param_resolver_local.cc | 42 DeviceResolverInterface* dev_resolver, const string& task_name) in CollectiveParamResolverLocal() argument
45 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()
|
| D | ring_reducer_test.cc | 46 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
48 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| D | hierarchical_tree_broadcaster_test.cc | 138 FailTestRMA(const DeviceMgr* dev_mgr, DeviceResolverInterface* dev_resolver, in FailTestRMA() argument
140 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 229 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init() local
234 dev_resolver.get(), worker_cache, in Init()
237 config, worker_env_.device_mgr, std::move(dev_resolver), in Init()
|