Function BM_GraphViewGetFaninEdges

tensorflow/core/grappler/graph_view_test.cc:491–505

RUN_FANIN_FANOUT_BENCHMARK(BM_GraphViewGetFanoutEdges);

static void BM_GraphViewGetFaninEdges(int iters, int num_fanins,
                                      int num_fanouts) {
  testing::StopTiming();
  const GraphDef graph_def = test::CreateFaninFanoutNodeGraph(
      num_fanins, num_fanouts, num_fanins, num_fanouts,
      /*fanout_unique_index=*/true);
  GraphView graph_view(&graph_def);

  testing::StartTiming();
  for (int i = 0; i < iters; ++i) {
    const NodeDef* node = graph_view.GetNode("node");
    graph_view.GetFaninEdges(*node, /*include_controlling_edges=*/false);
  }
  testing::StopTiming();
}

RUN_FANIN_FANOUT_BENCHMARK(BM_GraphViewGetFaninEdges);