GitHub

tf.compat.v1.GraphDef

A protobuf containing the graph of operations.

Migrate to TF2

This API is not available in TensorFlow 2.x.

You should not need to use GraphDefs directly in TF2. To load GraphDefs in TF2, use SavedModel. The SavedModel contains the GraphDef.

Before:

with tf.io.gfile.GFile('/tmp/graph.pb', 'rb') as f:
  graph_def = tf.compat.v1.GraphDef()
  graph_def.ParseFromString(f.read())

After:

tf.saved_model.load('/tmp/saved_model')

If you would like to create a GraphDef in TF2, use tf.function and get_concrete_function.

@tf.function
def f(x):
  return x

graph_def = f.get_concrete_function(1.).graph.as_graph_def()
print(graph_def)

Description

Used in the notebooks

Used in the guide

debug_info GraphDebugInfo debug_info library FunctionDefLibrary library node repeated NodeDef node version int32 version versions VersionDef versions