tf.mlir.experimental.convert_function

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

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

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

tf.function можно импортировать и преобразовать из TensorFlow в TensorFlow MLIR с помощью этого API, извлекая его ConcreteFunction (wrapper, выполняющий eager-вычисления, вокруг 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 {...} {...}...'