tf.mlir.experimental.convert_function

Импортировать ConcreteFunction и преобразовать её в текстовый модуль MLIR.

tf.mlir.experimental.convert_function(
    concrete_function, pass_pipeline='tf-standard-pipeline'
)

Этот API предназначен только для проверки внутренних компонентов TensorFlow, и возвращаемая строка в настоящее время предназначена для отладки.

tf.function можно импортировать и преобразовать из TensorFlow в TensorFlow MLIR с помощью этого API, извлекая его ConcreteFunction (безотлагательный исполняемый оболочку вокруг tf.Graph).

Например:

@tf.function
def add(a, b):
  return a + b

concrete_function = add.get_concrete_function(
    tf.TensorSpec(None, tf.dtypes.float32),
    tf.TensorSpec(None, tf.dtypes.float32))
tf.mlir.experimental.convert_function(concrete_function)
'...module attributes {...} {...}'