diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2018-04-04 16:17:46 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-04 16:22:19 -0700 |
| commit | f8acfb01792886274778d9ad7a9d990cbef14141 (patch) | |
| tree | e089cae1d1813458fad1de0eff700d0d5ff57221 /tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc | |
| parent | e98c13c55e519cb70ede110cd8941f8cb75ab718 (diff) | |
Fixed handling of control dependencies in the arithmethic optimizer
PiperOrigin-RevId: 191665098
Diffstat (limited to 'tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc b/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc index 416327e622..3f5ab87a5a 100644 --- a/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc +++ b/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc @@ -58,8 +58,8 @@ TEST_F(GraphOptimizerStageTest, ParseNodeNameAndScope_InScope) { TEST_F(GraphOptimizerStageTest, OptimizedNodeName) { GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, /*optimized_graph*/ nullptr, - /*graph_properties*/ nullptr, /*node_name*/ nullptr, - /*frame_map*/ nullptr); + /*graph_properties*/ nullptr, + /*node_name*/ nullptr); FakeOptimizerStage stage("my_opt", "my_stg", ctx); const auto node = ParseNodeScopeAndName("a/b/c/Add"); @@ -94,8 +94,7 @@ TEST_F(GraphOptimizerStageTest, GetInputNodeAndProperties) { GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, /*optimized_graph*/ &item.graph, /*graph_properties*/ &properties, - /*node_name*/ &node_map, - /*frame_map*/ nullptr); + /*node_name*/ &node_map); FakeOptimizerStage stage("my_opt", "my_stg", ctx); NodeDef* add_node; @@ -134,8 +133,7 @@ TEST_F(GraphOptimizerStageTest, AddNodes) { GraphOptimizerContext ctx(/*nodes_to_preserve*/ nullptr, /*optimized_graph*/ &item.graph, /*graph_properties*/ &properties, - /*node_name*/ &node_map, - /*frame_map*/ nullptr); + /*node_name*/ &node_map); FakeOptimizerStage stage("my_opt", "my_stg", ctx); NodeDef* add_node; @@ -165,4 +163,4 @@ TEST_F(GraphOptimizerStageTest, AddNodes) { } // namespace } // end namespace grappler -} // end namespace tensorflow
\ No newline at end of file +} // end namespace tensorflow |