diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-09 10:43:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-09 10:47:17 -0700 |
commit | 6b51853e3ab388af8f56685450f3b6fa5eb54ced (patch) | |
tree | 987c2a5d840d56498af6be18109e87b8d37eca51 /tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc | |
parent | f83a382e87ca09e8f688515a9549c81d0f46554a (diff) |
Automated rollback of commit 6874e1ef40c4189d96c105227f60b507953f95d3
PiperOrigin-RevId: 203790544
Diffstat (limited to 'tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc')
-rw-r--r-- | tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc | 42 |
1 files changed, 0 insertions, 42 deletions
diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc b/tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc index 54fdc01adb..c387b00303 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc @@ -279,11 +279,6 @@ class ArithmeticOptimizerTest : public GrapplerTest { optimizer->options_.optimize_max_or_min_of_monotonic = true; } - void EnableOnlyExpm1(ArithmeticOptimizer* optimizer) { - DisableAllStages(optimizer); - optimizer->options_.convert_expm1 = true; - } - void EnableOnlyUnaryOpsComposition(ArithmeticOptimizer* optimizer) { DisableAllStages(optimizer); optimizer->options_.unary_ops_composition = true; @@ -2547,43 +2542,6 @@ TEST_F(ArithmeticOptimizerTest, Log1p) { CompareGraphs(want, got); } -TEST_F(ArithmeticOptimizerTest, Expm1) { - tensorflow::Scope s = tensorflow::Scope::NewRootScope(); - - auto x1 = ops::Const(s.WithOpName("x1"), {2.0f, 2.0f}, {1, 2}); - auto x2 = ops::Const(s.WithOpName("x2"), {1.0f, 1.0f}, {1, 2}); - auto x3 = ops::Const(s.WithOpName("x3"), {3.0f, 3.0f}, {1, 2}); - auto s12 = ops::Sub(s.WithOpName("s12").WithControlDependencies(x3), x1, x2); - auto s23 = ops::Sub(s.WithOpName("s23"), x2, x3); - Output out1 = ops::Exp(s.WithOpName("out1"), s12); - Output out2 = ops::Exp(s.WithOpName("out2"), s23); - - GrapplerItem item; - item.fetch = {"out1", "out2"}; - TF_CHECK_OK(s.ToGraphDef(&item.graph)); - auto tensors_expected = EvaluateNodes(item.graph, item.fetch); - EXPECT_EQ(2, tensors_expected.size()); - - GraphDef got; - ArithmeticOptimizer optimizer; - EnableOnlyExpm1(&optimizer); - OptimizeAndPrune(&optimizer, &item, &got); - auto tensors = EvaluateNodes(got, item.fetch); - EXPECT_EQ(2, tensors.size()); - - GraphDef want; - AddNode("x1", "Const", {}, {}, &want); - AddNode("x2", "Const", {}, {}, &want); - AddNode("x3", "Const", {}, {}, &want); - AddNode("s23", "Sub", {"x2", "x3"}, {}, &want); - AddNode("out1", "Expm1", - {"x1", AsControlDependency("x2"), AsControlDependency("x3")}, {}, - &want); - AddNode("out2", "Exp", {"s23"}, {}, &want); - - CompareGraphs(want, got); -} - TEST_F(ArithmeticOptimizerTest, MinimizeBroadcasts_SimpleSwap) { tensorflow::Scope s = tensorflow::Scope::NewRootScope(); |