Searched refs:EagerClient (Results 1 – 15 of 15) sorted by relevance
29 class EagerClient : public core::RefCounted {31 ~EagerClient() override {} in ~EagerClient()85 core::RefCountPtr<EagerClient>* client) = 0;
91 core::RefCountPtr<EagerClient> eager_client;96 EagerClient* eager_client, std::unique_ptr<EagerOperation> op) in FunctionData()99 eager_client(core::RefCountPtr<EagerClient>(eager_client)), in FunctionData()
43 uint64 context_view_id, EagerClient* eager_client, in RemoteExecuteNode()110 const EagerClient* eager_client() const override { return eager_client_; } in eager_client()131 EagerClient* eager_client_; // Not owned, and must outlive this node.
34 core::RefCountPtr<EagerClient> eager_client, in DestroyTensorHandleNode()81 core::RefCountPtr<EagerClient> eager_client_;
67 core::RefCountPtr<eager::EagerClient> eager_client; in Instantiate()155 EagerClient* eager_client = function_data->eager_client.get(); in Run()266 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp()
168 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend()230 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv()394 core::RefCountPtr<eager::EagerClient> eager_client; in StartSendPackedHandle()447 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor()
39 core::RefCountPtr<eager::EagerClient> eager_client; in DestroyRemoteTensorHandle()
74 class FakeEagerClient : public EagerClient {125 core::RefCountPtr<EagerClient>* client) override { in GetClient()132 core::RefCountPtr<EagerClient> client_;760 core::RefCountPtr<EagerClient> client; in Init()
44 class EagerClient; variable98 virtual const eager::EagerClient* eager_client() const = 0;259 const eager::EagerClient* last_eager_client_;
568 core::RefCountPtr<eager::EagerClient> client; in CloseRemoteContexts()852 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely()894 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers()1035 core::RefCountPtr<eager::EagerClient> eager_client; in SyncExecutors()1246 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()1251 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()1284 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()1643 core::RefCountPtr<eager::EagerClient> client; in SetMasterContextState()
390 core::RefCountPtr<eager::EagerClient>* client);392 core::RefCountPtr<eager::EagerClient>* client);394 core::RefCountPtr<eager::EagerClient>* client);
186 core::RefCountPtr<eager::EagerClient> eager_client; in GetReplacedFromExistingWorkers()233 core::RefCountPtr<eager::EagerClient> eager_client; in CreateRemoteContexts()330 core::RefCountPtr<eager::EagerClient> eager_client; in UpdateRemoteContexts()
1541 core::RefCountPtr<eager::EagerClient> eager_client; in EagerRemoteExecute()
116 class GrpcEagerClient : public EagerClient {279 core::RefCountPtr<EagerClient>* client) override { in GetClient()291 core::RefCountPtr<EagerClient> worker( in GetClient()323 std::unordered_map<string, core::RefCountPtr<EagerClient>> clients_
44 core::RefCountPtr<EagerClient> eager_client; in TEST()