Skip to content

aot_operator

Classes

AotOperator 

AotOperator(ident: int, op: OperatorT, model: ModelT, subgraph: SubGraphT, prefix: str = 'aot', attributes: dict[str, str] = {})

Base class for all AOT operators.

This class provides a common interface for all operators in the HeliosAOT framework.

Parameters:

  • ident

    (int) –

    Operator identifier.

  • op

    (OperatorT) –

    Operator object from LiteRT model.

  • model

    (ModelT) –

    LiteRT model object.

  • subgraph

    (SubGraphT) –

    Subgraph object from LiteRT model.

  • prefix

    (str, default: 'aot' ) –

    Prefix for generated files. Defaults to "aot".

  • attributes

    (dict[str, str], default: {} ) –

    Attributes for template values. Defaults to {}.

Attributes

type property 
type: str

Return the operator type.

name property 
name: str

Return the operator name.

input_indices property 
input_indices: list[int]

Return the input tensor indices.

output_indices property 
output_indices: list[int]

Return the output tensor indices.

input_tensors property 
input_tensors: list[TensorT]

Return the input tensors.

output_tensors property 
output_tensors: list[TensorT]

Return the output tensors.

Functions

validate 
validate() -> bool

Validate the operator. This method should be overridden by subclasses.

Returns:

  • bool ( bool ) –

    True if the operator configuration is valid/supported, False otherwise.

compute_scratch_size 
compute_scratch_size(mode: str | None = 'MVE') -> int

Compute the size of the scratch buffer.

Parameters:

  • mode
    (str | None, default: 'MVE' ) –

    Mode for buffer size calculation. Defaults to "MVE".

Returns:

  • int ( int ) –

    Size of the scratch buffer in bytes.

compute_values 
compute_values() -> dict[str, str]

Compute the values for the operator source code template.

Args:

Returns:

  • dict[str, str] –

    dict[str, str]: Dictionary of values for the operator template.

print_info 
print_info(verbose: int = 0)

Debug print the template values for the operator.

generate_source_code 
generate_source_code(save_path: Path)

Generate the source code for the operator.

This method should be overridden by subclasses.

Parameters:

  • save_path
    (Path) –

    Path to save the generated source code.

Functions