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

Attributes
library	FunctionDefLibrary library
node	repeated NodeDef node
version	int32 version
versions	VersionDef versions