tensorflow / 2.9.1 / summary / graph.html /

tf.summary.graph

Writes a TensorFlow graph summary.

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)
Args
graph_data The TensorFlow graph to write, as a tf.Graph or a tf.compat.v1.GraphDef.
Returns
True on success, or False if no summary was written because no default summary writer was available.
Raises
ValueError graph summary API is invoked in a graph mode.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/summary/graph