Searched refs:cluster_flr_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime_test.cc | 50 cluster_flr_.reset(new ClusterFunctionLibraryRuntime(worker_session_.get(), in ClusterFunctionLibraryRuntimeTest()
69 cluster_flr_->Instantiate(function_name, lib_def, attrs, options, in Instantiate()
81 cluster_flr_->Instantiate(function_name, lib_def, attrs, options, &handle, in InstantiateAndRun()
94 cluster_flr_->Run(opts, handle, args, &out, in InstantiateAndRun()
114 std::unique_ptr<ClusterFunctionLibraryRuntime> cluster_flr_; member in tensorflow::ClusterFunctionLibraryRuntimeTest
|
| D | worker_session.h | 56 return cluster_flr_.get(); in cluster_flr()
114 std::unique_ptr<ClusterFunctionLibraryRuntime> cluster_flr_; variable
|
| D | worker_session.cc | 116 cluster_flr_(new ClusterFunctionLibraryRuntime( in WorkerSession()
163 cluster_flr_(new ClusterFunctionLibraryRuntime( in WorkerSession()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 88 cluster_flr_(cluster_flr), in EagerContext()
327 cluster_flr_.Reset(cluster_flr, /*owned=*/true); in ResetClusterFLR()
1356 thread_pool_.get(), cluster_flr_.Get()); in SetMasterContextState()
1458 thread_pool_.get(), cluster_flr_.Get()); in InitializeRemoteWorker()
|
| D | context.h | 610 OwnedOrUnownedHelper<DistributedFunctionLibraryRuntime> cluster_flr_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 138 cluster_flr_.reset(new TestClusterFLR(device_mgr_.get())); in Init()
142 /*thread_pool=*/nullptr, cluster_flr_.get(), session_metadata, in Init()
327 std::unique_ptr<TestClusterFLR> cluster_flr_; member in tensorflow::__anon135d08d60111::ProcessFunctionLibraryRuntimeTest
|