Function CreateGraphDef

tensorflow/core/graph/benchmark_testlib.h:72–112

    .Output("y: float");

GraphDef CreateGraphDef(int num_nodes, int num_edges_per_node) {
  const int kNumInNodes = 10 * num_edges_per_node;
  GraphDef graph_def;

  auto create_node = [](const string& name, const string& op) {
    NodeDef node;
    node.set_name(name);
    node.set_op(op);
    return node;
  };

  NodeDef node;
  for (int in = 0; in < kNumInNodes; ++in) {
    node = create_node(/*name=*/absl::StrFormat("in%04d", in), /*op=*/"Input");
    *graph_def.add_node() = std::move(node);
  }

  random::PhiloxRandom philox(301, 17);
  random::SimplePhilox rnd(&philox);
  for (int op = 0; op < num_nodes; ++op) {
    node = create_node(/*name=*/absl::StrFormat("op%05d", op),
                       /*op=*/absl::StrFormat("In%dOut1", num_edges_per_node));
    for (int edge = 0; edge < num_edges_per_node; ++edge) {
      node.add_input(absl::StrFormat("in%04d", rnd.Uniform(kNumInNodes)));
    }
    *graph_def.add_node() = std::move(node);
  }

  // Add a single sink node. Otherwise a lot of time is spent in
  // FixupSourceAndSinkEdges().
  node = create_node(/*name=*/"out", /*op=*/"Output");
  for (int op = 0; op < num_nodes; ++op) {
    node.add_input(absl::StrFormat("op%05d", op));
  }
  AttrValue attr;
  attr.set_i(num_nodes);
  node.mutable_attr()->insert({"N", std::move(attr)});
  *graph_def.add_node() = std::move(node);

  return graph_def;
}

GraphDef CreateRandomGraph(int size) {
  random::PhiloxRandom philox(0x12345);