Searched refs:remote_worker (Results 1 – 6 of 6) 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 | 31 const string& remote_worker, WorkerCacheInterface* wc) in CompleteGroupCall() argument
32 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
55 const string& remote_worker, WorkerCacheInterface* wc) in CompleteInstanceCall() argument
56 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 167 for (const string& remote_worker : removed_remote_workers) { in RemoveRemoteDevicesFromMgr() local
168 if (tensorflow::DeviceNameUtils::IsSameAddressSpace(remote_worker, in RemoveRemoteDevicesFromMgr()
279 const string& remote_worker = remote_workers[i]; in CreateRemoteContexts() local
281 if (!tensorflow::DeviceNameUtils::ParseFullName(remote_worker, in CreateRemoteContexts()
284 "Unable to parse ", remote_worker, " as a device name"); in CreateRemoteContexts()
290 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
293 "Cannot find a client for the given target:", remote_worker); in CreateRemoteContexts()
313 remote_worker, base_request.cluster_device_attributes(), in CreateRemoteContexts()
373 const string& remote_worker = remote_workers[i]; in UpdateRemoteContexts() local
375 if (!tensorflow::DeviceNameUtils::ParseFullName(remote_worker, in UpdateRemoteContexts()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 834 const string& remote_worker, const std::vector<string>& device_filters) { in SetRemoteDeviceFilters() argument
838 if (!DeviceNameUtils::ParseFullName(remote_worker, &pw)) { in SetRemoteDeviceFilters()
840 "Remote worker task name is invalid ", remote_worker); in SetRemoteDeviceFilters()
848 "Job name and task index must be specified for worker ", remote_worker); in SetRemoteDeviceFilters()
862 VLOG(1) << "Setting device filters for " << remote_worker << ":"; in SetRemoteDeviceFilters()
873 const string& remote_worker, in FilterDevicesForRemoteWorkers() argument
880 auto it = cluster_device_filters_.find(remote_worker); in FilterDevicesForRemoteWorkers()
889 DeviceNameUtils::ParseFullName(remote_worker, &parsed_remote_worker); in FilterDevicesForRemoteWorkers()
|
| D | context.h | 367 Status SetRemoteDeviceFilters(const string& remote_worker,
380 const string& remote_worker,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | constant-fold.mlir | 211 …%2 = "tf.Add"(%0, %1) {device = "/job:remote_worker/replica:123/task:456/CPU:0", name = "add"} : (…
|