Home
last modified time | relevance | path

Searched refs:virtual_cluster_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc47 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 1)); in SetUp()
52 virtual_cluster_.reset( in SetUp()
55 TF_CHECK_OK(virtual_cluster_->Provision()); in SetUp()
58 void TearDown() override { TF_CHECK_OK(virtual_cluster_->Shutdown()); } in TearDown()
226 std::unique_ptr<Cluster> virtual_cluster_; member in tensorflow::grappler::__anon47ae83710111::LayoutOptimizerTest
239 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
274 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
295 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
308 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
321 Status status = optimizer.Optimize(virtual_cluster_.get(), item, &output); in TEST_F()
[all …]
Dgeneric_layout_optimizer_test.cc168 virtual_cluster_ = in SetUp()
184 virtual_cluster_ = in SetUp()
189 virtual_cluster_ = in SetUp()
193 TF_ASSERT_OK(virtual_cluster_->Provision()); in SetUp()
196 void TearDown() override { TF_ASSERT_OK(virtual_cluster_->Shutdown()); } in TearDown()
198 std::unique_ptr<Cluster> virtual_cluster_; member in tensorflow::grappler::GenericLayoutOptimizerTest
240 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
285 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
304 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
327 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
[all …]
Dauto_mixed_precision_test.cc108 virtual_cluster_.reset(new SingleMachine(/* timeout_s = */ 10, 1, 1)); in SetUp()
116 virtual_cluster_.reset( in SetUp()
119 TF_CHECK_OK(virtual_cluster_->Provision()); in SetUp()
122 void TearDown() override { TF_CHECK_OK(virtual_cluster_->Shutdown()); } in TearDown()
186 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TestSimpleUnaryInferOp()
205 std::unique_ptr<Cluster> virtual_cluster_; member in tensorflow::grappler::__anond48c34ab0111::AutoMixedPrecisionTest
225 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
263 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
306 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
351 TF_ASSERT_OK(optimizer.Optimize(virtual_cluster_.get(), item, &output)); in TEST_F()
[all …]
Dgeneric_layout_optimizer_transposer_test.cc359 virtual_cluster_ = in SetUp()
365 virtual_cluster_ = in SetUp()
368 TF_ASSERT_OK(virtual_cluster_->Provision()); in SetUp()
371 void TearDown() override { TF_ASSERT_OK(virtual_cluster_->Shutdown()); } in TearDown()
401 item, virtual_cluster_.get(), &context)); in ReduceTransposerKeepDims()
468 item, virtual_cluster_.get(), &context)); in ReduceTransposerValidAxisNode()
503 std::unique_ptr<Cluster> virtual_cluster_; member in tensorflow::grappler::__anon90176d1d0111::TransposerTest
514 item, virtual_cluster_.get(), &context)); in TEST_F()
558 item, virtual_cluster_.get(), &context)); in TEST_F()
598 item, virtual_cluster_.get(), &context)); in TEST_F()
[all …]