onnx.inliner

inline_local_functions

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

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

参数:
  • model – ONNX ModelProto

  • convert_version – 如果为真,则尝试对需要与模型不同的 (ONNX) 操作集版本的函数应用自动版本转换。

返回值:

所有对模型本地函数的调用都已内联(递归)的 ModelProto

inline_selected_functions

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

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

参数:
  • model – ONNX ModelProto

  • function_ids – 要在内联时包含/排除的函数列表。每个元素都是一个 (函数域、函数名) 元组。

  • exclude – 如果为真,则内联除 function_ids 中指定的函数之外的所有函数。如果为假,则内联 function_ids 中指定的所有函数。

返回值:

所有对模型本地函数的调用都已内联(递归)的 ModelProto