Python tf.mlir.experimental.convert_function用法及代码示例

导入 ConcreteFunction 并将其转换为文本 MLIR 模块。

用法

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

参数

concrete_function ConcreteFunction 类型的对象。
pass_pipeline 要在模块上运行的 MLIR Pass Pipeline 的文本说明，请参阅 MLIR 文档文本传递管道语法.
show_debug_info 是否在发出的文本形式中包含位置。

对应于 ConcreteFunction 的 MLIR 模块的文本表示。

抛出

InvalidArgumentError 如果 concrete_function 无效或无法转换为 MLIR。

此 API 仅用于检查 TensorFlow 的内部，返回的字符串目前用于调试目的。

通过提取其 ConcreteFunction(围绕 tf.Graph 的eagerly-executing 包装器)，可以使用此 API 导入 tf.function 并将其从 TensorFlow 转换为 TensorFlow MLIR。

例如：

@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 {...} {...}...'

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.mlir.experimental.convert_function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。