Function TEST_F

tensorflow/core/grappler/utils/pattern_utils_test.cc:107–139

class PatternMatcherTest : public ::testing::Test {};

TEST_F(PatternMatcherTest, MatMulBiasAddGeluNodeIndices) {
  ::tensorflow::Status status;
  GraphDef graph;
  GetMatMulBiasAddGeluGraph(&graph);
  OpTypePattern pattern = GetMatMulBiasAddGeluPattern();
  MutableGraphView graph_view(&graph, &status);
  TF_ASSERT_OK(status);
  graph_view.SortTopologically(/*ignore_cycles=*/false, {});
  auto root_node_view = graph_view.GetNode("gelu");

  SubGraphMatcher<MatchingDirection::kFollowInputs> graph_matcher(&graph_view);
  std::map<string, int> matched_nodes_map;  // label to node index map
  std::set<int> remove_node_indices;
  bool found_match = graph_matcher.GetMatchedNodes(
      pattern, root_node_view, &matched_nodes_map, &remove_node_indices);

  EXPECT_TRUE(found_match);
  EXPECT_FALSE(matched_nodes_map.empty());
  EXPECT_FALSE(remove_node_indices.empty());

  bool all_indices_matched = true;
  for (auto it = matched_nodes_map.begin(); it != matched_nodes_map.begin();
       it++) {
    auto label = absl::StripPrefix(it->first, "my_");
    int matched_node_idx = it->second;
    int expected_node_idx = graph_view.GetNode(label)->node_index();
    if (matched_node_idx != expected_node_idx) {
      all_indices_matched = false;
      break;
    }
  }
  EXPECT_TRUE(all_indices_matched);
}

// Pattern should not be matched if any of candidate remove nodes has external
// dependent.