tf.compat.v1.GraphDef

Протобуффер, содержащий граф операций.

Этот API недоступен в TensorFlow 2.x.

Вам не должно потребоваться использовать GraphDef напрямую в TF2. Для загрузки GraphDef в TF2 используйте SavedModel. SavedModel содержит GraphDef.

До:

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

После:

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

Если вы хотите создать GraphDef в TF2, используйте tf.function и get_concrete_function.

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

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

Описание