| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 52 cluster_.reset( in SetUp()
54 TF_CHECK_OK(cluster_->EnablePeakMemoryStats()); in SetUp()
55 TF_CHECK_OK(cluster_->Provision()); in SetUp()
59 if (cluster_) { in TearDown()
60 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
62 cluster_.reset(); in TearDown()
66 std::unique_ptr<SingleMachine> cluster_; member in tensorflow::grappler::__anon6fff17260111::SingleMachineTest
70 CHECK_EQ("single_machine", cluster_->type()); in TEST_F()
75 cluster_->GetDeviceNames()); in TEST_F()
79 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
[all …]
|
| D | virtual_cluster_test.cc | 47 cluster_.reset(new VirtualCluster(devices)); in SetUp()
48 TF_CHECK_OK(cluster_->Provision()); in SetUp()
52 TF_CHECK_OK(cluster_->Shutdown()); in TearDown()
53 cluster_.reset(); in TearDown()
57 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::__anonc76e83b80111::VirtualClusterTest
61 CHECK_EQ("virtual", cluster_->type()); in TEST_F()
66 cluster_->GetDeviceNames()); in TEST_F()
70 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
73 TF_CHECK_OK(cluster_->Run(item.graph, item.feed, item.fetch, &metadata)); in TEST_F()
116 TF_CHECK_OK(cluster_->Initialize(item)); in TEST_F()
[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 | measuring_cost_estimator.h | 65 Cluster* cluster_; // Not owned.
|
| D | graph_properties_test.cc | 64 cluster_.reset(new SingleMachine(5 * 60, 3, 0)); in SetUp()
65 TF_ASSERT_OK(cluster_->Provision()); in SetUp()
93 TF_ASSERT_OK(cluster_->Shutdown()); in TearDown()
94 cluster_.reset(); in TearDown()
149 std::unique_ptr<SingleMachine> cluster_; member in tensorflow::grappler::__anon66ac7c850111::GraphPropertiesTest
155 cluster_->GetDeviceNames()); in TEST_F()
196 cluster_->GetDeviceNames()); in TEST_F()
223 cluster_->GetDeviceNames()); in TEST_F()
228 TF_ASSERT_OK(cluster_->Initialize(item)); in TEST_F()
229 Status s = properties.InferDynamically(cluster_.get()); in TEST_F()
[all …]
|
| D | virtual_scheduler.h | 380 Cluster* GetCluster() { return cluster_; } in GetCluster()
439 Cluster* cluster_; // Not owned. variable
|
| D | virtual_scheduler_test.cc | 677 cluster_ = absl::make_unique<VirtualCluster>(devices); in VirtualSchedulerTest()
681 cluster_.get()); in VirtualSchedulerTest()
2382 std::unique_ptr<VirtualCluster> cluster_; member in tensorflow::grappler::__anon653cd6ac0111::VirtualSchedulerTest
3078 cluster_.get()); in TEST_F()
|
| D | virtual_scheduler.cc | 389 cluster_(cluster), in SchedulerState()
427 TF_RETURN_IF_ERROR(graph_properties_->InferDynamically(cluster_)); in Init()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device_test.cc | 41 std::unique_ptr<test::TestCluster> cluster_; member in tensorflow::RemoteDeviceTest
47 TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 1, &cluster_)); in RemoteDeviceTest()
48 const string& hostport = cluster_->targets()[0]; in RemoteDeviceTest()
|
| D | cluster_function_library_runtime_test.cc | 34 TF_CHECK_OK(test::TestCluster::MakeTestCluster(options, 2, &cluster_)); in ClusterFunctionLibraryRuntimeTest()
36 TF_CHECK_OK(spec.AddHostPortsJob("localhost", cluster_->targets())); in ClusterFunctionLibraryRuntimeTest()
112 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()
54 cluster_->targets()[0], &options.config.rpc_options(), &channel_ptr)); in MasterTest()
58 std::unique_ptr<test::TestCluster> cluster_; member in tensorflow::MasterTest
|
| /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::__anon61f011e50111::SCCTest
57 cluster_->GetDeviceNames()); in TEST_F()
|
| /external/rust/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/lb_policy/xds/ |
| D | cds.cc | 49 explicit CdsLbConfig(std::string cluster) : cluster_(std::move(cluster)) {} in CdsLbConfig()
50 const std::string& cluster() const { return cluster_; } in cluster()
54 std::string cluster_; member in grpc_core::__anon5aa779ee0111::CdsLbConfig
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 49 R"CODE( subgraph "cluster_%s" { style=rounded bgcolor="%s" penwidth=0.0 label=%s
|