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:

• AirActivationType ( AirActivationType ) –

  The corresponding AIR activation type.

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_depth_to_space

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

Parse a depth to space 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_sub

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

Parse a subtraction 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_to_space_nd

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

Parse a batch to space 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_reduce_max

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

Parse a REDUCE_MAX 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_reduce_min

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

Parse a REDUCE_MIN 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_arg_max

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

Parse an ARG_MAX operator.

parse_arg_min

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

Parse an ARG_MIN 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_space_to_batch_nd

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

Parse a space to batch 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_space_to_depth

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

Parse a space to depth 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_slice

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

Parse a slice 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_gather

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

Parse a gather operator.

parse_gather_nd

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

Parse a gather_nd 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_ethos_u

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

Parse an Ethos-U custom 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.

parse_svdf

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

Parse an SVDF 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_prelu

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

Parse a PReLU 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.