Searched refs:remote_worker (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 122 for (const string& remote_worker : removed_remote_workers) { in RemoveRemoteDevicesFromMgr() local
123 if (DeviceNameUtils::IsSameAddressSpace(remote_worker, d->name())) { in RemoveRemoteDevicesFromMgr()
224 const string& remote_worker = remote_workers[i]; in CreateRemoteContexts() local
226 if (!DeviceNameUtils::ParseFullName(remote_worker, &parsed_name)) { in CreateRemoteContexts()
227 statuses[i] = errors::InvalidArgument("Unable to parse ", remote_worker, in CreateRemoteContexts()
234 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
237 "Cannot find a client for the given target:", remote_worker); in CreateRemoteContexts()
256 remote_worker, base_request.cluster_device_attributes(), in CreateRemoteContexts()
321 const string& remote_worker = remote_workers[i]; in UpdateRemoteContexts() local
323 if (!DeviceNameUtils::ParseFullName(remote_worker, &parsed_name)) { in UpdateRemoteContexts()
[all …]
|
| D | context.cc | 893 for (auto& remote_worker : remote_workers) { in RegisterExistingFunctionsOnRemoteWorkers() local
895 Status s = GetClient(remote_worker, &eager_client); in RegisterExistingFunctionsOnRemoteWorkers()
1372 const string& remote_worker, const std::vector<string>& device_filters) { in SetRemoteDeviceFilters() argument
1376 if (!DeviceNameUtils::ParseFullName(remote_worker, &pw)) { in SetRemoteDeviceFilters()
1378 "Remote worker task name is invalid ", remote_worker); in SetRemoteDeviceFilters()
1386 "Job name and task index must be specified for worker ", remote_worker); in SetRemoteDeviceFilters()
1400 VLOG(1) << "Setting device filters for " << remote_worker << ":"; in SetRemoteDeviceFilters()
1411 const string& remote_worker, in FilterDevicesForRemoteWorkers() argument
1418 auto it = cluster_device_filters_.find(remote_worker); in FilterDevicesForRemoteWorkers()
1427 DeviceNameUtils::ParseFullName(remote_worker, &parsed_remote_worker); in FilterDevicesForRemoteWorkers()
|
| D | context.h | 464 Status SetRemoteDeviceFilters(const string& remote_worker,
477 const string& remote_worker,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() argument
34 remote_worker_(remote_worker), in CancellableCall()
|
| D | collective_param_resolver_distributed.cc | 37 const string& remote_worker, WorkerCacheInterface* wc) in CompleteGroupCall() argument
38 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
60 const string& remote_worker, WorkerCacheInterface* wc) in CompleteInstanceCall() argument
61 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
|