Searched refs:remote_eager_workers_ (Results 1 – 2 of 2) sorted by relevance
177 remote_eager_workers_->GetClient(worker_and_context_id.first); in CloseRemoteContexts()323 remote_eager_workers_->GetClient(target_and_context_id.first); in MaybeRegisterFunctionRemotely()410 *client = remote_eager_workers_->GetClient(device_task_name); in GetClientAndContextID()495 remote_eager_workers_ = std::move(remote_eager_workers); in InitializeRemote()533 auto* client = remote_eager_workers_->GetClient( in InitializeRemote()
323 std::unique_ptr<eager::EagerClientCache> remote_eager_workers_; variable