aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-09 10:43:22 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-09 10:47:17 -0700
commit6b51853e3ab388af8f56685450f3b6fa5eb54ced (patch)
tree987c2a5d840d56498af6be18109e87b8d37eca51 /tensorflow/core/grappler/optimizers/arithmetic_optimizer_test.cc
parentf83a382e87ca09e8f688515a9549c81d0f46554a (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.cc42
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();