Optimization Examples

Examples demonstrating the optimization features of onnx-light exposed by onnx_light.onnx_optim (shape inference, expression helpers, …).

ComputeContext memory expressions

ComputeContext memory expressions

Evaluating inferred shapes with concrete input dimensions

Evaluating inferred shapes with concrete input dimensions

Optimized Shape inference

Optimized Shape inference

Shape inference with a custom operator

Shape inference with a custom operator

Symbolic expressions for dimensions

Symbolic expressions for dimensions

pretty_onnx: shape info, shape tags, inplace and release annotations

pretty_onnx: shape info, shape tags, inplace and release annotations