tf.io.write_graph

Записывает протокол графа в файл.

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

Граф записывается как текстовый протокол, если as_text False.

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

или

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