Function write_graph

tensorflow/compiler/aot/tests/make_test_graphs.py:172–179

Build a graph using build_graph and write it out.

(build_graph, out_dir)

def write_graph(build_graph, out_dir):
  """Build a graph using build_graph and write it out."""
  g = ops.Graph()
  with g.as_default():
    build_graph(out_dir)
    filename = os.path.join(out_dir, 'test_graph_%s.pb' % build_graph.__name__)
    with open(filename, 'wb') as f:
      f.write(g.as_graph_def().SerializeToString())


def main(_):