furiosa.optimizer.frontend.onnx.utils package

Submodules

furiosa.optimizer.frontend.onnx.utils.check_model module

furiosa.optimizer.frontend.onnx.utils.check_model.check_model(model: ModelProto, check_runnable: bool = True) → None

Check if model’s well-defined and executable on onnxruntime

furiosa.optimizer.frontend.onnx.utils.inference_shape module

class furiosa.optimizer.frontend.onnx.utils.inference_shape.InferenceShape(model: ModelProto)

Bases: object

Replace former InferenceShape with ONNX_Simplifier https://github.com/daquexian/onnx-simplifier

inference_shape(input_shapes: Mapping[str, List[int | None]] | None = None) → ModelProto

furiosa.optimizer.frontend.onnx.utils.version_checker module

class furiosa.optimizer.frontend.onnx.utils.version_checker.CheckVersion(opset_version: int | None = None)

Bases: Transformer[ModelProto]

Convert an ONNX model to ONNX opset 12 or 13

opset_bound = [12, 13]

transform(model: ModelProto) → ModelProto

Module contents