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, а возвращаемая строка в настоящее время предназначена для отладки.

С помощью этого API объект tf.function можно импортировать и преобразовать из TensorFlow в TensorFlow MLIR, извлекая его 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 {...} {...}...'