diff options
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc')
-rw-r--r-- | tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc b/tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc index a46c504ac4..b676246b31 100644 --- a/tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc +++ b/tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc @@ -85,8 +85,8 @@ TEST(MapAndBatchFusionTest, FuseMapAndBatchNodesIntoOne) { EXPECT_FALSE( graph_utils::ContainsGraphNodeWithName(batch_node->name(), output)); EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDatasetV2", output)); - NodeDef map_and_batch_node = - output.node(graph_utils::FindNodeWithOp("MapAndBatchDatasetV2", output)); + NodeDef map_and_batch_node = output.node( + graph_utils::FindGraphNodeWithOp("MapAndBatchDatasetV2", output)); EXPECT_EQ(map_and_batch_node.input_size(), 5); EXPECT_EQ(map_and_batch_node.input(0), map_node->input(0)); EXPECT_EQ(map_and_batch_node.input(1), map_node->input(1)); @@ -170,8 +170,8 @@ TEST(MapAndBatchFusionTest, FuseMapAndBatchV2NodesIntoOne) { EXPECT_FALSE( graph_utils::ContainsGraphNodeWithName(batch_node->name(), output)); EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDatasetV2", output)); - NodeDef map_and_batch_node = - output.node(graph_utils::FindNodeWithOp("MapAndBatchDatasetV2", output)); + NodeDef map_and_batch_node = output.node( + graph_utils::FindGraphNodeWithOp("MapAndBatchDatasetV2", output)); EXPECT_EQ(map_and_batch_node.input_size(), 5); EXPECT_EQ(map_and_batch_node.input(0), map_node->input(0)); EXPECT_EQ(map_and_batch_node.input(1), map_node->input(1)); @@ -253,8 +253,8 @@ TEST(MapAndBatchFusionTest, FuseParallelMapAndBatchNodesIntoOne) { EXPECT_FALSE( graph_utils::ContainsGraphNodeWithName(batch_node->name(), output)); EXPECT_TRUE(graph_utils::ContainsNodeWithOp("MapAndBatchDatasetV2", output)); - NodeDef map_and_batch_node = - output.node(graph_utils::FindNodeWithOp("MapAndBatchDatasetV2", output)); + NodeDef map_and_batch_node = output.node( + graph_utils::FindGraphNodeWithOp("MapAndBatchDatasetV2", output)); EXPECT_EQ(map_and_batch_node.input_size(), 5); EXPECT_EQ(map_and_batch_node.input(0), map_node->input(0)); EXPECT_EQ(map_and_batch_node.input(1), map_node->input(1)); |