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) → ModelProto[來源]

在給定的模型中嵌入選取的模型本機函式。

參數:

  • model – 一個 ONNX ModelProto

  • function_ids – 要在嵌入時包含/排除的函式清單。每個元素都是 (函式網域、函式名稱) 的元組。

  • exclude – 若為 true，則嵌入 function_ids 中指定的所有函式，若為 false，則嵌入 function_ids 中指定的所有函式。

傳回:

ModelProto，其中所有對模型本機函式的呼叫都已嵌入 (遞迴)