Home
last modified time | relevance | path

Searched refs:EagerClient (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_client.h30 class EagerClient : public core::RefCounted {
32 ~EagerClient() override {} in ~EagerClient()
85 core::RefCountPtr<EagerClient>* client) = 0;
Dcluster_function_library_runtime.h89 core::RefCountPtr<EagerClient> eager_client;
94 EagerClient* eager_client, std::unique_ptr<EagerOperation> op) in FunctionData()
97 eager_client(core::RefCountPtr<EagerClient>(eager_client)), in FunctionData()
Dremote_execute_node.h42 uint64 context_view_id, EagerClient* eager_client, in RemoteExecuteNode()
109 const EagerClient* eager_client() const override { return eager_client_; } in eager_client()
130 EagerClient* eager_client_; // Not owned, and must outlive this node.
Ddestroy_tensor_handle_node.h33 core::RefCountPtr<EagerClient> eager_client, in DestroyTensorHandleNode()
80 core::RefCountPtr<EagerClient> eager_client_;
Dcluster_function_library_runtime.cc68 core::RefCountPtr<eager::EagerClient> eager_client; in Instantiate()
156 EagerClient* eager_client = function_data->eager_client.get(); in Run()
265 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp()
Dremote_copy_node.cc164 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend()
222 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv()
385 core::RefCountPtr<eager::EagerClient> eager_client; in StartSendPackedHandle()
439 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor()
Dremote_tensor_handle_data.cc38 core::RefCountPtr<eager::EagerClient> eager_client; in DestroyRemoteTensorHandle()
Deager_service_impl_test.cc69 class FakeEagerClient : public EagerClient {
119 core::RefCountPtr<EagerClient>* client) override { in GetClient()
126 core::RefCountPtr<EagerClient> client_;
754 core::RefCountPtr<EagerClient> client; in Init()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_executor.h44 class EagerClient; variable
98 virtual const eager::EagerClient* eager_client() const = 0;
257 const eager::EagerClient* last_eager_client_;
Dcontext.cc457 core::RefCountPtr<eager::EagerClient> client; in CloseRemoteContexts()
638 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely()
679 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers()
820 core::RefCountPtr<eager::EagerClient> eager_client; in SyncExecutors()
993 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
998 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
1031 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
1385 core::RefCountPtr<eager::EagerClient> client; in SetMasterContextState()
Dcontext.h341 core::RefCountPtr<eager::EagerClient>* client);
343 core::RefCountPtr<eager::EagerClient>* client);
345 core::RefCountPtr<eager::EagerClient>* client);
Dcontext_distributed_manager.cc187 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in GetReplacedFromExistingWorkers()
235 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in CreateRemoteContexts()
333 tensorflow::core::RefCountPtr<tensorflow::eager::EagerClient> eager_client; in UpdateRemoteContexts()
Dexecute.cc832 core::RefCountPtr<eager::EagerClient> eager_client; in EagerRemoteExecute()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc107 class GrpcEagerClient : public EagerClient {
248 core::RefCountPtr<EagerClient>* client) override { in GetClient()
260 core::RefCountPtr<EagerClient> worker( in GetClient()
292 std::unordered_map<string, core::RefCountPtr<EagerClient>> clients_
Dgrpc_eager_client_test.cc45 core::RefCountPtr<EagerClient> eager_client; in TEST()