onnx.inliner¶
inline_local_functions¶
- onnx.inliner.inline_local_functions(model: ModelProto, convert_version: bool = False) ModelProto [来源]¶
在给定模型中内联模型局部函数。
- 参数:
model – 一个 ONNX ModelProto
convert_version – 如果为 true,尝试对需要不同(ONNX)opset 版本(与模型不同)的函数应用自动版本转换。
- 返回:
ModelProto,其中所有对模型局部函数的调用都已内联(递归地)
inline_selected_functions¶
- onnx.inliner.inline_selected_functions(model: ModelProto, function_ids: list[tuple[str, str]], exclude: bool = False, inline_schema_functions: bool = False) ModelProto [来源]¶
在给定模型中内联选定的函数。
- 参数:
model – 一个 ONNX ModelProto
function_ids – 内联时要包含/排除的函数列表。每个元素是 (函数域, 函数名) 的元组。
exclude – 如果为 true,则内联除 function_ids 中指定的所有函数。如果为 false,则内联 function_ids 中指定的所有函数。
inline_schema_functions – 如果为 true,则内联 schema 定义的函数以及模型局部函数。否则,只内联模型局部函数。
- 返回:
ModelProto,其中所有对模型局部函数的调用都已内联(递归地)