Searched refs:remote_eager_workers (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 217 eager::EagerClientCache* remote_eager_workers, in CreateRemoteContexts() argument
234 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in CreateRemoteContexts()
296 eager::EagerClientCache* remote_eager_workers, in UpdateRemoteContexts() argument
331 statuses[i] = remote_eager_workers->GetClient(remote_worker, &eager_client); in UpdateRemoteContexts()
453 std::unique_ptr<eager::EagerClientCache> remote_eager_workers; in UpdateContextWithServerDef() local
456 &remote_eager_workers)); in UpdateContextWithServerDef()
506 remote_eager_workers.get(), &replaced_workers)); in UpdateContextWithServerDef()
554 server_def, remote_eager_workers.get(), context->Executor().Async(), in UpdateContextWithServerDef()
579 keep_alive_secs, server_def, remote_eager_workers.get(), in UpdateContextWithServerDef()
595 remote_eager_workers.get(), base_request)); in UpdateContextWithServerDef()
[all …]
|
| D | context.cc | 1460 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in InitializeRemoteMaster() argument
1483 std::move(remote_eager_workers), std::move(remote_device_manager), in InitializeRemoteMaster()
1490 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in UpdateRemoteMaster() argument
1528 remote_eager_workers_ = std::move(remote_eager_workers); in UpdateRemoteMaster()
1558 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in SetMasterContextState() argument
1596 remote_eager_workers_ = std::move(remote_eager_workers); in SetMasterContextState()
1678 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in InitializeRemoteWorker() argument
1707 remote_eager_workers_ = std::move(remote_eager_workers); in InitializeRemoteWorker()
1734 std::unique_ptr<eager::EagerClientCache> remote_eager_workers, in UpdateRemoteWorker() argument
1747 remote_eager_workers_ = std::move(remote_eager_workers); in UpdateRemoteWorker()
|
| D | context.h | 420 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
436 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
443 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
456 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
774 std::unique_ptr<eager::EagerClientCache> remote_eager_workers,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 296 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers; in CreateContext() local
298 &remote_eager_workers)); in CreateContext()
305 std::move(remote_eager_workers), worker_session->remote_device_mgr(), in CreateContext()
422 std::unique_ptr<tensorflow::eager::EagerClientCache> remote_eager_workers; in UpdateContext() local
424 &remote_eager_workers)); in UpdateContext()
427 Status s = ctx->UpdateRemoteWorker(std::move(remote_eager_workers), in UpdateContext()
|