aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc')
-rw-r--r--tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc b/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc
index 34f28c7c27..799c40c67b 100644
--- a/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc
+++ b/tensorflow/core/grappler/optimizers/graph_optimizer_stage_test.cc
@@ -61,6 +61,7 @@ TEST_F(GraphOptimizerStageTest, OptimizedNodeName) {
/*optimized_graph*/ nullptr,
/*graph_properties*/ nullptr,
/*node_name*/ nullptr,
+ /*feed_nodes*/ nullptr,
/*opt_level*/ RewriterConfig::ON);
FakeOptimizerStage stage("my_opt", "my_stg", ctx);
@@ -97,6 +98,7 @@ TEST_F(GraphOptimizerStageTest, GetInputNodeAndProperties) {
/*optimized_graph*/ &item.graph,
/*graph_properties*/ &properties,
/*node_name*/ &node_map,
+ /*feed_nodes*/ nullptr,
/*opt_level*/ RewriterConfig::ON);
FakeOptimizerStage stage("my_opt", "my_stg", ctx);
@@ -137,6 +139,7 @@ TEST_F(GraphOptimizerStageTest, AddNodes) {
/*optimized_graph*/ &item.graph,
/*graph_properties*/ &properties,
/*node_name*/ &node_map,
+ /*feed_nodes*/ nullptr,
/*opt_level*/ RewriterConfig::ON);
FakeOptimizerStage stage("my_opt", "my_stg", ctx);