autocast

Modules

modelopt.onnx.autocast.convert(onnx_path[, ...])

Convert model to mixed precision.

modelopt.onnx.autocast.graphsanitizer

Graph sanitization and optimization for ONNX models.

modelopt.onnx.autocast.logging_config

Logging configuration module for AutoCast.

modelopt.onnx.autocast.nodeclassifier

Node classification module for AutoCast.

modelopt.onnx.autocast.precisionconverter

Precision conversion module for ONNX models.

modelopt.onnx.autocast.referencerunner

Reference runner module for ONNX model execution.

modelopt.onnx.autocast.utils

Utility functions for AutoCast.

AutoCast package for converting FP32 ONNX models to mixed precision.