tf.compat.v1.GraphDef

Protobuf, содержащий граф операций.

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

В TF2 вам не потребуется напрямую использовать GraphDefs. Чтобы загружать GraphDefs в 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)

Описание

Используется в блокнотах