tf.summary.graph

View source on GitHub

Writes a TensorFlow graph summary.

tf.summary.graph(
    graph_data
)

Write an instance of tf.Graph or tf.compat.v1.GraphDef as summary only in an eager mode. Please prefer to use the trace APIs (tf.summary.trace_on, tf.summary.trace_off, and tf.summary.trace_export) when using tf.function which can automatically collect and record graphs from executions.

Usage Example:

writer = tf.summary.create_file_writer("/tmp/mylogs")

@tf.function
def f():
  x = constant_op.constant(2)
  y = constant_op.constant(3)
  return x**y

with writer.as_default():
  tf.summary.graph(f.get_concrete_function().graph)

# Another example: in a very rare use case, when you are dealing with a TF v1
# graph.
graph = tf.Graph()
with graph.as_default():
  c = tf.constant(30.0)
with writer.as_default():
  tf.summary.graph(graph)