Skip to content

op_parsers

Classes

Functions

register_op_parser 

register_op_parser(op_type: str) -> Callable[[OpParserFn], OpParserFn]

Decorator to register an operator parser for a specific operator type.

Parameters:

  • op_type

    (str) –

    The operator type to register the parser for.

Returns:

  • Callable[[OpParserFn], OpParserFn] –

    Callable[[OpParserFn], OpParserFn]: The decorator function.

get_op_parser 

get_op_parser(op_type: str) -> OpParserFn | None

Get the operator parser for a specific operator type.

Parameters:

  • op_type

    (str) –

    The LiteRT operator type

Returns:

  • OpParserFn | None –

    OpParserFn | None: The operator parser function, or None if not found.

parse_tensor_type 

parse_tensor_type(tensor_type: TensorType) -> AirTensorType

Convert a LiteRT TensorType to AIR TensorType.

Parameters:

  • tensor_type

    (TensorType) –

    The LiteRT tensor type.

Returns:

  • AirTensorType ( AirTensorType ) –

    The corresponding AIR tensor type.

parse_activation_type 

parse_activation_type(value: ActivationFunctionType) -> AirActivationType

Convert a LiteRT ActivationFunctionType to AIR ActivationType.

Parameters:

  • value

    (ActivationFunctionType) –

    The LiteRT activation function type.

Returns:

parse_conv_2d 

parse_conv_2d(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D convolution operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_depthwise_conv_2d 

parse_depthwise_conv_2d(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a depthwise 2D convolution operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_transpose_conv 

parse_transpose_conv(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D transpose convolution operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_add 

parse_add(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D addition operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_assign_variable 

parse_assign_variable(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a variable assignment operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_average_pool_2d 

parse_average_pool_2d(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D average pooling operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_batch_matmul 

parse_batch_matmul(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a batch matrix multiplication operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_concatenation 

parse_concatenation(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D concatenation operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_dequantize 

parse_dequantize(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a dequantization operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_expand_dims 

parse_expand_dims(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse an expand dimensions operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_fill 

parse_fill(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a fill operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_fully_connected 

parse_fully_connected(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a fully connected operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_hard_swish 

parse_hard_swish(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a hard swish operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_leaky_relu 

parse_leaky_relu(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a leaky ReLU operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_logistic 

parse_logistic(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a logistic (sigmoid) operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_max_pool_2d 

parse_max_pool_2d(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a 2D max pooling operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_maximum 

parse_maximum(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a maximum operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_mean 

parse_mean(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a mean operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_minimum 

parse_minimum(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a minimum operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_mul 

parse_mul(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a multiplication operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_pack 

parse_pack(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a pack operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_pad 

parse_pad(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a padding operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_quantize 

parse_quantize(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a quantization operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_read_variable 

parse_read_variable(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a variable read operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_relu 

parse_relu(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a ReLU operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_reshape 

parse_reshape(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a reshape operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_shape 

parse_shape(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a shape operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_softmax 

parse_softmax(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a softmax operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_split 

parse_split(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a split operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_split_v 

parse_split_v(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a split_v operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_squeeze 

parse_squeeze(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a squeeze operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_strided_slice 

parse_strided_slice(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a strided slice operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_tanh 

parse_tanh(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a tanh operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_transpose 

parse_transpose(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a transpose operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_unpack 

parse_unpack(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse an unpack operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.

parse_zeros_like 

parse_zeros_like(id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel) -> AirOperator

Parse a zeros_like operator.

Parameters:

  • id

    (int) –

    The operator ID.

  • model

    (ModelT) –

    The LiteRT model.

  • subgraph

    (SubGraphT) –

    The LiteRT subgraph.

  • air_model

    (AirModel) –

    The AIR model.

Returns:

  • AirOperator ( AirOperator ) –

    The parsed AIR operator.