| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.cc | 68 core::RefCountPtr<eager::EagerClient> eager_client; in Instantiate() local
69 s = ctx_->GetClient(target, &eager_client); in Instantiate()
75 if (eager_client == nullptr) { in Instantiate()
100 eager_client->EnqueueAsync( in Instantiate()
103 target, ret_indices, eager_client = eager_client.get(), in Instantiate()
108 function_data_.emplace_back(target, ret_indices, eager_client, in Instantiate()
156 EagerClient* eager_client = function_data->eager_client.get(); in Run() local
157 if (eager_client == nullptr) { in Run()
222 eager_client->RunComponentFunctionAsync( in Run()
265 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp() local
[all …]
|
| D | cluster_function_library_runtime.h | 89 core::RefCountPtr<EagerClient> eager_client; member
94 EagerClient* eager_client, std::unique_ptr<EagerOperation> op) in FunctionData()
97 eager_client(core::RefCountPtr<EagerClient>(eager_client)), in FunctionData()
99 eager_client->Ref(); in FunctionData()
|
| D | remote_copy_node.cc | 164 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend() local
165 status = ctx_->GetClient(send_device_, &eager_client); in StartSend()
176 eager_client->StreamingEnqueueAsync( in StartSend()
222 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv() local
223 Status status = ctx_->GetClient(recv_device_, &eager_client); in RunRemoteRecv()
245 eager_client->StreamingEnqueueAsync( in RunRemoteRecv()
385 core::RefCountPtr<eager::EagerClient> eager_client; in StartSendPackedHandle() local
386 s = ctx_->GetClient(recv_device_, &eager_client); in StartSendPackedHandle()
396 eager_client->StreamingEnqueueAsync( in StartSendPackedHandle()
439 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor() local
[all …]
|
| D | BUILD | 38 ":eager_client",
61 ":eager_client",
71 name = "eager_client",
72 hdrs = ["eager_client.h"],
86 ":eager_client",
205 ":eager_client",
238 "eager_client.h",
|
| D | remote_execute_node.h | 42 uint64 context_view_id, EagerClient* eager_client, in RemoteExecuteNode() argument
52 eager_client_(eager_client), in RemoteExecuteNode()
109 const EagerClient* eager_client() const override { return eager_client_; } in eager_client() function
|
| D | destroy_tensor_handle_node.h | 33 core::RefCountPtr<EagerClient> eager_client, in DestroyTensorHandleNode() argument
37 eager_client_(std::move(eager_client)), in DestroyTensorHandleNode()
|
| D | remote_tensor_handle_data.cc | 38 core::RefCountPtr<eager::EagerClient> eager_client; in DestroyRemoteTensorHandle() local
39 Status status = ctx->GetClient(remote_task, &eager_client); in DestroyRemoteTensorHandle()
57 std::move(request), std::move(eager_client), ready)); in DestroyRemoteTensorHandle()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_client_test.cc | 45 core::RefCountPtr<EagerClient> eager_client; in TEST() local
46 Status s = client_cache->GetClient(target, &eager_client); in TEST()
|
| D | BUILD | 32 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 187 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in GetReplacedFromExistingWorkers() local
189 client_cache->GetClient(existing_workers->at(i), &eager_client); in GetReplacedFromExistingWorkers()
194 eager_client->KeepAliveAsync( in GetReplacedFromExistingWorkers()
235 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in CreateRemoteContexts() local
236 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
237 if (eager_client == nullptr) { in CreateRemoteContexts()
275 eager_client->CreateContextAsync( in CreateRemoteContexts()
333 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in UpdateRemoteContexts() local
334 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in UpdateRemoteContexts()
335 if (eager_client == nullptr) { in UpdateRemoteContexts()
[all …]
|
| D | eager_executor.cc | 361 async_remote_node->eager_client() != nullptr && in RunItem()
362 last_eager_client_ != async_remote_node->eager_client()) { in RunItem()
373 if (async_remote_node->eager_client() != nullptr && in RunItem()
378 last_eager_client_ = async_remote_node->eager_client(); in RunItem()
|
| D | context.cc | 638 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely() local
639 TF_RETURN_IF_ERROR(GetClient(target, &eager_client)); in MaybeRegisterFunctionRemotely()
642 eager_client->StreamingEnqueueAsync( in MaybeRegisterFunctionRemotely()
679 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers() local
680 Status s = GetClient(remote_worker, &eager_client); in RegisterExistingFunctionsOnRemoteWorkers()
686 eager_client->StreamingEnqueueAsync( in RegisterExistingFunctionsOnRemoteWorkers()
820 core::RefCountPtr<eager::EagerClient> eager_client; in SyncExecutors() local
821 TF_RETURN_IF_ERROR(GetClient(target, &eager_client)); in SyncExecutors()
824 eager_client->StreamingEnqueueAsync( in SyncExecutors()
|
| D | BUILD | 120 "//tensorflow/core/distributed_runtime/eager:eager_client",
218 "//tensorflow/core/distributed_runtime/eager:eager_client",
532 "//tensorflow/core/distributed_runtime/eager:eager_client",
656 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| D | eager_executor.h | 98 virtual const eager::EagerClient* eager_client() const = 0;
|
| D | execute.cc | 832 core::RefCountPtr<eager::EagerClient> eager_client; in EagerRemoteExecute() local
834 TF_RETURN_IF_ERROR(ctx.GetClient(op->GetDeviceParsedName(), &eager_client)); in EagerRemoteExecute()
961 ctx.GetContextViewId(), eager_client.get(), op->GetCancellationManager(), in EagerRemoteExecute()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 95 "//tensorflow/core/distributed_runtime/eager:eager_client",
993 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | BUILD | 281 "//tensorflow/core/distributed_runtime/eager:eager_client",
|