| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.cc | 53 core::RefCountPtr<eager::EagerClient> eager_client; in Instantiate() local
60 s = ctx_->GetClient(device, &eager_client); in Instantiate()
66 if (eager_client == nullptr) { in Instantiate()
89 eager_client->EnqueueAsync( in Instantiate()
92 target, eager_client = eager_client.get(), done](const Status& s) { in Instantiate()
96 function_data_.emplace_back(target, eager_client, in Instantiate()
124 EagerClient* eager_client = function_data->eager_client.get(); in Run() local
125 if (eager_client == nullptr) { in Run()
167 eager_client->EnqueueAsync(request, response, in Run()
188 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp() local
[all …]
|
| D | cluster_function_library_runtime.h | 73 core::RefCountPtr<EagerClient> eager_client; member
76 FunctionData(const string& target, EagerClient* eager_client, in FunctionData()
79 eager_client(core::RefCountPtr<EagerClient>(eager_client)), in FunctionData()
81 eager_client->Ref(); in FunctionData()
|
| D | remote_copy_node.cc | 159 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend() local
160 status = ctx_->GetClient(send_device_, &eager_client); in StartSend()
171 eager_client->StreamingEnqueueAsync( in StartSend()
202 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv() local
203 Status status = ctx_->GetClient(recv_device_, &eager_client); in RunRemoteRecv()
225 eager_client->StreamingEnqueueAsync( in RunRemoteRecv()
308 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor() local
309 s = ctx_->GetClient(recv_device_, &eager_client); in StartRemoteSendTensor()
319 eager_client->StreamingEnqueueAsync( in StartRemoteSendTensor()
|
| D | BUILD | 33 ":eager_client",
53 ":eager_client",
63 name = "eager_client",
64 hdrs = ["eager_client.h"],
77 ":eager_client",
192 ":eager_client",
223 "eager_client.h",
|
| D | destroy_tensor_handle_node.h | 33 EagerClient* eager_client, bool ready) in DestroyTensorHandleNode() argument
36 eager_client_(eager_client), in DestroyTensorHandleNode()
|
| D | remote_tensor_handle_data.cc | 37 core::RefCountPtr<eager::EagerClient> eager_client; in DestroyRemoteTensorHandle() local
38 Status status = ctx->GetClient(remote_task, &eager_client); in DestroyRemoteTensorHandle()
56 std::move(request), eager_client.get(), ready)); in DestroyRemoteTensorHandle()
|
| D | remote_execute_node.h | 40 EagerClient* eager_client, const NodeDef& ndef, in RemoteExecuteNode() argument
47 eager_client_(eager_client), in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 240 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in GetReplacedFromExistingWorkers() local
242 client_cache->GetClient(existing_workers->at(i), &eager_client); in GetReplacedFromExistingWorkers()
247 eager_client->KeepAliveAsync( in GetReplacedFromExistingWorkers()
289 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in CreateRemoteContexts() local
290 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
291 if (eager_client == nullptr) { in CreateRemoteContexts()
328 eager_client->CreateContextAsync( in CreateRemoteContexts()
383 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in UpdateRemoteContexts() local
384 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in UpdateRemoteContexts()
385 if (eager_client == nullptr) { in UpdateRemoteContexts()
[all …]
|
| D | BUILD | 74 "//tensorflow/core/distributed_runtime/eager:eager_client",
251 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | BUILD | 31 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 79 "//tensorflow/core/distributed_runtime/eager:eager_client",
167 "//tensorflow/core/distributed_runtime/eager:eager_client",
356 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| D | context.cc | 525 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely() local
526 TF_RETURN_IF_ERROR(remote_eager_workers_->GetClient(target, &eager_client)); in MaybeRegisterFunctionRemotely()
529 eager_client->StreamingEnqueueAsync( in MaybeRegisterFunctionRemotely()
551 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers() local
553 remote_eager_workers_->GetClient(remote_workers[i], &eager_client); in RegisterExistingFunctionsOnRemoteWorkers()
564 eager_client->StreamingEnqueueAsync( in RegisterExistingFunctionsOnRemoteWorkers()
|
| D | execute.cc | 640 core::RefCountPtr<eager::EagerClient> eager_client; in EagerRemoteExecute() local
642 TF_RETURN_IF_ERROR(ctx.GetClient(op->GetDeviceParsedName(), &eager_client)); in EagerRemoteExecute()
773 std::move(request), op_device, eager_client.get(), in EagerRemoteExecute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | BUILD | 278 "//tensorflow/core/distributed_runtime/eager:eager_client",
|