Skip to content

transform_pipeline

Classes

TransformPipeline 

TransformPipeline(transforms: list[AotTransform])

Holds a sequence of AotTransform instances and applies them in order.

Parameters:

Functions

apply 
apply(model: ModelT, subgraph: SubGraphT, operations: list[Type[AotOperator]]) -> tuple[litert.ModelT, litert.SubGraphT, list[Type[AotOperator]]]

Apply the sequence of transforms to the model, subgraph, and operations. Args: model (litert.ModelT): The model to transform. subgraph (litert.SubGraphT): The subgraph to transform. operations (list[Type["AotOperator"]]): The list of operations to transform.

Returns:

  • tuple[ModelT, SubGraphT, list[Type[AotOperator]]] –

    tuple[litert.ModelT, litert.SubGraphT, list[Type["AotOperator"]]]: The transformed model, subgraph, and operations.

from_config classmethod 
from_config(cfg: TransformsConfig) -> TransformPipeline

Build a pipeline from a TransformsConfig, preserving the registry order and enabling/disabling as per cfg.transforms.

Parameters:

  • cfg
    (TransformsConfig) –

    Configuration object containing transform settings.

Returns:

  • TransformPipeline ( TransformPipeline ) –

    An instance of TransformPipeline with the specified transforms.

Functions

apply_wildcard_and_validate 

apply_wildcard_and_validate(cfg: TransformsConfig) -> TransformsConfig

Apply wildcard and validate the transforms configuration.

Parameters:

  • cfg

    (TransformsConfig) –

    Configuration object containing transform settings.

Returns:

  • TransformsConfig ( TransformsConfig ) –

    Updated configuration object with validated transforms.