tf.compat.v1.GraphDef

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)