Kano001's picture
Upload 2707 files
dc2106c verified
raw
history blame
780 Bytes
# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
def inline_local_functions(model: bytes, convert_version: bool) -> bytes:
"""Inlines calls to model-local function in input model and returns it.
Both input and output are serialized ModelProtos.
"""
...
def inline_selected_functions(model: bytes, function_ids: list[tuple[str,str]], exclude: bool) -> bytes:
"""Inlines calls to selected model-local functions in input model and returns it.
Inlines all functions specified in function_ids, unless exclude is true, in which
case it inlines all functions except those specified in function_ids.
Both input and output are serialized ModelProtos.
"""
...