Home
last modified time | relevance | path

Searched refs:remote_eager_workers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc178 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers; in CreateContext() local
180 &remote_eager_workers)); in CreateContext()
187 std::move(remote_eager_workers), worker_session->remote_device_mgr(), in CreateContext()
284 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers; in UpdateContext() local
286 &remote_eager_workers)); in UpdateContext()
293 device_mgr, std::move(remote_eager_workers), in UpdateContext()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h318 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
334 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
343 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
357 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
558 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
Dcontext.cc916 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in InitializeRemoteMaster() argument
936 std::move(remote_eager_workers), std::move(remote_device_manager), in InitializeRemoteMaster()
943 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in UpdateRemoteMaster() argument
986 remote_eager_workers_ = std::move(remote_eager_workers); in UpdateRemoteMaster()
1020 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in SetMasterContextState() argument
1051 remote_eager_workers_ = std::move(remote_eager_workers); in SetMasterContextState()
1133 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in InitializeRemoteWorker() argument
1162 remote_eager_workers_ = std::move(remote_eager_workers); in InitializeRemoteWorker()
1190 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in UpdateRemoteWorker() argument
1207 remote_eager_workers_ = std::move(remote_eager_workers); in UpdateRemoteWorker()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc272 tensorflow::eager::EagerClientCache* remote_eager_workers, bool async, in CreateRemoteContexts() argument
290 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
348 tensorflow::eager::EagerClientCache* remote_eager_workers, in UpdateRemoteContexts() argument
384 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in UpdateRemoteContexts()
503 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers; in UpdateTFE_ContextWithServerDef() local
506 &remote_eager_workers)); in UpdateTFE_ContextWithServerDef()
545 remote_eager_workers.get(), &replaced_workers)); in UpdateTFE_ContextWithServerDef()
596 server_def, remote_eager_workers.get(), context->Executor().Async(), in UpdateTFE_ContextWithServerDef()
606 server_def, remote_eager_workers.get(), context->Executor().Async(), in UpdateTFE_ContextWithServerDef()
616 context_view_id + 1, server_def, remote_eager_workers.get(), in UpdateTFE_ContextWithServerDef()
[all …]