Searched refs:eager_client (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 33 EagerClient* eager_client, in RemoteExecuteNode() argument
39 eager_client_(eager_client), in RemoteExecuteNode()
49 EagerClient* eager_client) in RemoteExecuteNode() argument
52 eager_client_(eager_client) {} in RemoteExecuteNode()
|
| D | BUILD | 24 name = "eager_client",
25 hdrs = ["eager_client.h"],
36 ":eager_client",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 529 EagerContext* ctx, eager::EagerClient* eager_client, uint64 context_id, in GetRemoteTensorDestructor() argument
531 return [ctx, eager_client, context_id, op_id, output_num]() { in GetRemoteTensorDestructor()
549 new eager::RemoteExecuteNode(id, std::move(request), eager_client); in GetRemoteTensorDestructor()
554 eager_client->EnqueueAsync( in GetRemoteTensorDestructor()
581 eager::EagerClient* eager_client; in EagerRemoteSendTensor()
584 ctx->GetClientAndContextID(recv_device, &eager_client, &context_id)); in EagerRemoteSendTensor()
617 eager_client->SendTensorAsync(&request, &response, in EagerRemoteSendTensor()
626 GetRemoteTensorDestructor(ctx, eager_client, context_id, id, 0); in EagerRemoteSendTensor()
648 eager::EagerClient* eager_client; in EagerRemoteExecute()
651 ctx->GetClientAndContextID(op->Device(), &eager_client, &context_id)); in EagerRemoteExecute()
[all …]
|
| D | BUILD | 70 "//tensorflow/core/distributed_runtime/eager:eager_client",
230 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| D | context.cc | 322 auto* eager_client = in MaybeRegisterFunctionRemotely() local
325 eager_client->RegisterFunctionAsync( in MaybeRegisterFunctionRemotely()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 63 "//tensorflow/core/distributed_runtime/eager:eager_client",
105 "//tensorflow/core/distributed_runtime/eager:eager_client",
204 "//tensorflow/core/distributed_runtime/eager:eager_client",
|
| D | c_api.cc | 135 auto* eager_client = remote_eager_workers->GetClient(remote_worker); in CreateRemoteContexts() local
136 if (eager_client == nullptr) { in CreateRemoteContexts()
143 eager_client->CreateContextAsync( in CreateRemoteContexts()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | BUILD | 27 "//tensorflow/core/distributed_runtime/eager:eager_client",
|