Searched refs:cluster_flr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 102 DistributedFunctionLibraryRuntime* cluster_flr = nullptr);
372 DistributedFunctionLibraryRuntime* cluster_flr,
396 DistributedFunctionLibraryRuntime* cluster_flr,
531 DistributedFunctionLibraryRuntime* cluster_flr = nullptr);
533 void ResetClusterFLR(DistributedFunctionLibraryRuntime* cluster_flr);
674 DistributedFunctionLibraryRuntime* cluster_flr,
|
| D | context.cc | 80 DistributedFunctionLibraryRuntime* cluster_flr) in EagerContext() argument
88 cluster_flr_(cluster_flr), in EagerContext()
102 thread_pool_.get(), cluster_flr);
190 DistributedFunctionLibraryRuntime* cluster_flr) { in ResetPFLR() argument
198 thread_pool, cluster_flr, in ResetPFLR()
326 DistributedFunctionLibraryRuntime* cluster_flr) { in ResetClusterFLR() argument
327 cluster_flr_.Reset(cluster_flr, /*owned=*/true); in ResetClusterFLR()
1207 DistributedFunctionLibraryRuntime* cluster_flr, in InitializeRemoteMaster() argument
1227 context_id, 0, r, local_device_mgr, keep_alive_secs, cluster_flr, in InitializeRemoteMaster()
1304 int keep_alive_secs, DistributedFunctionLibraryRuntime* cluster_flr, in SetMasterContextState() argument
[all …]
|
| D | context_distributed_manager.cc | 621 tensorflow::DistributedFunctionLibraryRuntime* cluster_flr = in UpdateContextWithServerDef() local
630 remote_workers, context_id, r, device_mgr, keep_alive_secs, cluster_flr, in UpdateContextWithServerDef()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 81 DistributedFunctionLibraryRuntime* cluster_flr,
185 DistributedFunctionLibraryRuntime* cluster_flr, Item* item);
|
| D | graph_mgr.cc | 129 DistributedFunctionLibraryRuntime* cluster_flr, Item* item) { in InitItem() argument
142 graph_options.optimizer_options(), worker_env_->compute_pool, cluster_flr, in InitItem()
299 DistributedFunctionLibraryRuntime* cluster_flr, string* graph_handle) { in Register() argument
302 config_proto, collective_graph_key, cluster_flr, item); in Register()
|
| D | worker_session.h | 55 ClusterFunctionLibraryRuntime* cluster_flr() const { in cluster_flr() function
|
| D | worker.cc | 85 session->cluster_flr(), response->mutable_graph_handle()); in RegisterGraphAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 277 request->async(), device_mgr, false, r, worker_session->cluster_flr()); in CreateContext()
291 DistributedFunctionLibraryRuntime* cluster_flr = in CreateContext() local
299 std::move(rendezvous_creator), cluster_flr, std::move(remote_mgr), in CreateContext()
|