onnx.inliner

内联本地函数

onnx.inliner.inline_local_functions(model: ModelProto, convert_version: bool = False) → ModelProto[source]

在给定模型中内联模型局部函数。

Parameters:

  • model – 一个ONNX ModelProto

  • convert_version – 如果为true，尝试对需要与模型不同（ONNX）操作集版本的函数应用自动版本转换。

Returns:

ModelProto，其中所有对模型本地函数的调用都已内联（递归地）

内联选择函数

onnx.inliner.inline_selected_functions(model: ModelProto, function_ids: list[tuple[str, str]], exclude: bool = False) → ModelProto[source]

在给定模型中内联选定的模型本地函数。

Parameters:

  • model – 一个ONNX ModelProto

  • function_ids – 内联时要包含/排除的函数列表。每个元素是一个元组，包含（函数域，函数名称）。

  • exclude – 如果为true，则内联所有函数，除了function_ids中指定的那些。 如果为false，则内联function_ids中指定的所有函数。

Returns:

ModelProto，其中所有对模型本地函数的调用都已内联（递归地）