tf.train.write_graph

tf.train.write_graph(
    graph_or_graph_def,
    logdir,
    name,
    as_text=True
)

Defined in tensorflow/python/framework/graph_io.py.

Writes a graph proto to a file.

The graph is written as a text proto unless as_text is False.

v = tf.Variable(0, name='my_variable')
sess = tf.Session()
tf.train.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')

or

v = tf.Variable(0, name='my_variable')
sess = tf.Session()
tf.train.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')

Args:

  • graph_or_graph_def: A Graph or a GraphDef protocol buffer.
  • logdir: Directory where to write the graph. This can refer to remote filesystems, such as Google Cloud Storage (GCS).
  • name: Filename for the graph.
  • as_text: If True, writes the graph as an ASCII proto.

Returns:

The path of the output proto file.