Searched refs:remote_eager_workers (Results 1 – 3 of 3) sorted by relevance
116 tensorflow::eager::EagerClientCache* remote_eager_workers, bool async, in CreateRemoteContexts() argument135 auto* eager_client = remote_eager_workers->GetClient(remote_worker); in CreateRemoteContexts()203 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers( in UpdateTFE_ContextWithServerDef() local210 remote_eager_workers.get(), ctx->context.Async(), &remote_contexts)); in UpdateTFE_ContextWithServerDef()230 std::move(server), std::move(remote_eager_workers), in UpdateTFE_ContextWithServerDef()
461 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in InitializeRemote() argument495 remote_eager_workers_ = std::move(remote_eager_workers); in InitializeRemote()
220 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,