| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 46 cluster_.reset( in SetUp()
48 TF_CHECK_OK(cluster_->EnablePeakMemoryStats(true)); in SetUp()
49 TF_CHECK_OK(cluster_->Provision()); in SetUp()
53 if (cluster_) { in TearDown()
54 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
56 cluster_.reset(); in TearDown()
60 std::unique_ptr<SingleMachine> cluster_; member in tensorflow::grappler::__anon53a42bef0111::SingleMachineTest
64 CHECK_EQ("single_machine", cluster_->type()); in TEST_F()
69 cluster_->GetDeviceNames()); in TEST_F()
73 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
[all …]
|
| D | virtual_cluster_test.cc | 46 cluster_.reset(new VirtualCluster(devices)); in SetUp()
47 TF_CHECK_OK(cluster_->Provision()); in SetUp()
51 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
52 cluster_.reset(); in TearDown()
56 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::__anon1fb631a10111::VirtualClusterTest
60 CHECK_EQ("virtual", cluster_->type()); in TEST_F()
65 cluster_->GetDeviceNames()); in TEST_F()
69 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
72 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
115 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 23 std::shared_ptr<tensorflow::grappler::Cluster> cluster_; member
117 GCluster(tensorflow::grappler::Cluster* cluster) : cluster_(cluster) {} in GCluster()
120 return cluster_.get();
123 return cluster_.get(); in get()
126 return cluster_.get() == nullptr; in is_none()
129 std::shared_ptr<tensorflow::grappler::Cluster> cluster_; member
138 tensorflow::grappler::Cluster* cluster_ = in TF_NewCluster() local
141 cluster_->DisableDetailedStats(disable_detailed_stats); in TF_NewCluster()
142 cluster_->AllowSoftPlacement(allow_soft_placement); in TF_NewCluster()
143 cluster_->SetNumWarmupSteps(10); in TF_NewCluster()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | measuring_cost_estimator.cc | 44 cluster_ = cluster; in MeasuringCostEstimator()
50 return cluster_->Initialize(item); in Initialize()
60 const bool running_simulation = (cluster_->type() == "virtual"); in PredictCosts()
73 cluster_->Run(optimized_graph, feed_, fetch_, &metadata); in PredictCosts()
|
| D | analytical_cost_estimator_test.cc | 46 cluster_.reset(new VirtualCluster(devices)); in SetUp()
91 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::AnalyticalCostEstimatorTest
97 AnalyticalCostEstimator estimator(cluster_.get(), /*use_static_shapes=*/true, in TEST_F()
|
| D | analytical_cost_estimator.cc | 120 : cluster_(cluster), in AnalyticalCostEstimator()
126 use_static_shapes_, use_aggressive_shape_inference_, cluster_, in AnalyticalCostEstimator()
|
| D | graph_properties_test.cc | 46 cluster_.reset(new SingleMachine(5 * 60, 3, 0)); in SetUp()
47 TF_CHECK_OK(cluster_->Provision()); in SetUp()
75 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
76 cluster_.reset(); in TearDown()
118 std::unique_ptr<SingleMachine> cluster_; member in tensorflow::grappler::__anonda2d9c2e0111::GraphPropertiesTest
124 cluster_->GetDeviceNames()); in TEST_F()
168 cluster_->GetDeviceNames()); in TEST_F()
195 cluster_->GetDeviceNames()); in TEST_F()
200 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
201 Status s = properties.InferDynamically(cluster_.get()); in TEST_F()
[all …]
|
| D | measuring_cost_estimator.h | 65 Cluster* cluster_; // Not owned.
|
| D | analytical_cost_estimator.h | 66 Cluster* cluster_;
|
| D | virtual_scheduler.h | 351 Cluster* cluster_; // Not owned. variable
|
| D | virtual_scheduler.cc | 265 cluster_(cluster), in VirtualScheduler()
304 TF_RETURN_IF_ERROR(graph_properties_->InferDynamically(cluster_)); in Init()
|
| D | virtual_scheduler_test.cc | 72 cluster_ = absl::make_unique<VirtualCluster>(devices); in VirtualSchedulerTest()
75 /*use_aggressive_shape_inference=*/true, cluster_.get()); in VirtualSchedulerTest()
1708 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::VirtualSchedulerTest
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | scc_test.cc | 34 cluster_.reset(new VirtualCluster(devices)); in SetUp()
35 TF_CHECK_OK(cluster_->Provision()); in SetUp()
38 void TearDown() override { cluster_.reset(); } in TearDown()
51 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::__anon59b28aee0111::SCCTest
57 cluster_->GetDeviceNames()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device_test.cc | 41 std::unique_ptr<test::TestCluster> cluster_; member in tensorflow::RemoteDeviceTest
46 TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 1, &cluster_)); in RemoteDeviceTest()
47 const string& hostport = cluster_->targets()[0]; in RemoteDeviceTest()
|
| D | cluster_function_library_runtime_test.cc | 33 TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_)); in ClusterFunctionLibraryRuntimeTest()
35 TF_CHECK_OK(spec.AddHostPortsJob("localhost", cluster_->targets())); in ClusterFunctionLibraryRuntimeTest()
99 std::unique_ptr<test::TestCluster> cluster_; member in tensorflow::ClusterFunctionLibraryRuntimeTest
|
| D | master_test.cc | 51 TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_)); in MasterTest()
53 TF_CHECK_OK(NewHostPortGrpcChannel(cluster_->targets()[0], &channel_ptr)); in MasterTest()
57 std::unique_ptr<test::TestCluster> cluster_; member in tensorflow::MasterTest
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 49 R"CODE( subgraph "cluster_%s" { style=rounded bgcolor="%s" penwidth=0.0 label=%s
|