Searched refs:remote_worker (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() argument
33 remote_worker_(remote_worker), in CancellableCall()
|
| D | collective_param_resolver_distributed.cc | 29 const string& remote_worker, WorkerCacheInterface* wc) in CompleteGroupCall() argument
30 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
52 const string& remote_worker, WorkerCacheInterface* wc) in CompleteInstanceCall() argument
53 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
|
| D | remote_device.h | 44 const string& remote_worker, NewRemoteDevicesDone done);
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 87 for (const string& remote_worker : remote_workers) { in GetAllRemoteDevices() local
90 tensorflow::Env::Default(), worker_cache, remote_worker, in GetAllRemoteDevices()
119 const string& remote_worker = remote_workers[i]; in CreateRemoteContexts() local
125 if (!tensorflow::DeviceNameUtils::ParseFullName(remote_worker, in CreateRemoteContexts()
128 "Unable to parse ", remote_worker, " as a device name"); in CreateRemoteContexts()
135 auto* eager_client = remote_eager_workers->GetClient(remote_worker); in CreateRemoteContexts()
138 "Cannot find a client for the given target:", remote_worker); in CreateRemoteContexts()
151 remote_contexts->emplace(remote_worker, response.context_id()); in CreateRemoteContexts()
|