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

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