Searched refs:DeviceResolverDistributed (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed.cc | 21 DeviceResolverDistributed::DeviceResolverDistributed( in DeviceResolverDistributed() function in tensorflow::DeviceResolverDistributed
26 void DeviceResolverDistributed::GetLocalityAsync(const string& device, in GetLocalityAsync()
67 void DeviceResolverDistributed::GetDeviceLocalitiesAsync( in GetDeviceLocalitiesAsync()
74 void DeviceResolverDistributed::GetDeviceLocalitiesRecursive( in GetDeviceLocalitiesRecursive()
96 void DeviceResolverDistributed::RefreshRemoteAttributes( in RefreshRemoteAttributes()
117 void DeviceResolverDistributed::ClearTask(const string& task) { in ClearTask()
|
| D | device_resolver_distributed.h | 29 class DeviceResolverDistributed : public DeviceResolverInterface {
31 DeviceResolverDistributed(const DeviceMgr* dev_mgr,
35 virtual ~DeviceResolverDistributed() {} in ~DeviceResolverDistributed()
|
| D | device_resolver_distributed_test.cc | 33 class TestableDeviceResolverDistributed : public DeviceResolverDistributed {
38 : DeviceResolverDistributed(dev_mgr, worker_cache, task) {} in TestableDeviceResolverDistributed()
65 DeviceResolverDistributed* dres) in FakeWorker()
82 DeviceResolverDistributed* device_resolver_;
185 DeviceResolverDistributed* dres = it.second; in TEST_F()
|
| D | collective_rma_distributed_test.cc | 64 DeviceResolverDistributed* dres) in FakeWorker()
121 DeviceResolverDistributed* device_resolver_;
226 DeviceResolverDistributed* dev_res = in DefineWorker()
227 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
245 std::unordered_map<string, DeviceResolverDistributed*> dev_resolvers_;
|
| D | rpc_collective_executor_mgr.h | 25 class DeviceResolverDistributed; variable
39 std::unique_ptr<DeviceResolverDistributed> dev_resolver,
|
| D | collective_param_resolver_distributed.h | 23 class DeviceResolverDistributed; variable
30 DeviceResolverDistributed* dev_resolver,
|
| D | rpc_collective_executor_mgr_test.cc | 48 std::unique_ptr<DeviceResolverDistributed> dr(new DeviceResolverDistributed( in RpcCollectiveExecutorMgrTest()
|
| D | collective_param_resolver_distributed_test.cc | 171 DeviceResolverDistributed* dev_res = in DefineWorker()
172 new DeviceResolverDistributed(dev_mgr, &wc_, worker_name); in DefineWorker()
286 std::unordered_map<string, DeviceResolverDistributed*> dev_resolvers_;
|
| D | rpc_collective_executor_mgr.cc | 30 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in RpcCollectiveExecutorMgr()
|
| D | collective_param_resolver_distributed.cc | 83 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_server_lib.cc | 71 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init()
72 new DeviceResolverDistributed(env->device_mgr, worker_cache, in Init()
|
| D | gdr_collective_executor_mgr.h | 37 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in GdrCollectiveExecutorMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 229 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init()
230 new DeviceResolverDistributed(worker_env_.device_mgr, worker_cache, in Init()
|