Searched defs:eager_client (Results 1 – 10 of 10) sorted by relevance
168 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend() local230 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv() local394 core::RefCountPtr<eager::EagerClient> eager_client; in StartSendPackedHandle() local447 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor() local
67 core::RefCountPtr<eager::EagerClient> eager_client; in Instantiate() local155 EagerClient* eager_client = function_data->eager_client.get(); in Run() local266 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp() local
43 uint64 context_view_id, EagerClient* eager_client, in RemoteExecuteNode()110 const EagerClient* eager_client() const override { return eager_client_; } in eager_client() function
34 core::RefCountPtr<EagerClient> eager_client, in DestroyTensorHandleNode()
91 core::RefCountPtr<EagerClient> eager_client; member
39 core::RefCountPtr<eager::EagerClient> eager_client; in DestroyRemoteTensorHandle() local
186 core::RefCountPtr<eager::EagerClient> eager_client; in GetReplacedFromExistingWorkers() local233 core::RefCountPtr<eager::EagerClient> eager_client; in CreateRemoteContexts() local330 core::RefCountPtr<eager::EagerClient> eager_client; in UpdateRemoteContexts() local
852 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely() local894 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers() local1035 core::RefCountPtr<eager::EagerClient> eager_client; in SyncExecutors() local
1541 core::RefCountPtr<eager::EagerClient> eager_client; in EagerRemoteExecute() local
44 core::RefCountPtr<EagerClient> eager_client; in TEST() local