transforms

Graph transform API.

The default registry orders static-shape folding, invariant-variable folding, eligible depthwise-to-conv rewrites, identity pruning, eligible transpose-convolution rewrites and, last, scheduling each DEQUANTIZE of a constant just before its first consumer. TransformPipeline expands settings in that registry order; configuration enables/disables named passes and supplies options without reordering them.

See TransformSpec, TransformPipeline and register_default_transforms below for the public contracts. Individual implementation modules are source code details rather than additional promised API pages.

Re-exports

Name	Source
AirTransform	helia_aot.transforms.transform.AirTransform
NoOptions	helia_aot.transforms.transform.NoOptions
TransformPipeline	helia_aot.transforms.transform_pipeline.TransformPipeline
TransformSpec	helia_aot.transforms.defines.TransformSpec

Machine-readable model

• register_default_transformsfunctionRegister all built-in transforms into the provided registry.

function

register_default_transforms

Python

Register all built-in transforms into the provided registry.

helia_aot/transforms/__init__.py:28

register_default_transforms(registry: Registry[str, type[AirTransform]]) -> None

Register all built-in transforms into the provided registry.