op_parsers

Classes

Functions

register_op_parser

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

Decorator to annotate default operator parser registrations.

get_op_parser

get_op_parser(registry: Registry[str | AirOpType, OpParserFn], op_type: str | AirOpType) -> OpParserFn | None

Get the operator parser for a specific operator type.

invoke_op_parser

invoke_op_parser(op_parser: OpParserFn, op_id: int, model: ModelT, subgraph: SubGraphT, air_model: AirModel, *, subgraph_index: int) -> AirOperator

Invoke a LiteRT op parser, forwarding the selected subgraph index when supported.

register_default_litert_parsers

register_default_litert_parsers(registry: Registry[str | AirOpType, OpParserFn]) -> None

Register all built-in LiteRT op parsers into the provided registry.

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_squared_difference

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

Parse a squared difference 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_broadcast_to

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

Parse a broadcast_to operator.

parse_dynamic_update_slice

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

Parse a dynamic update slice 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_sum

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

Parse a sum 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_select_v2

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

Parse a SELECT_V2 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_tile

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

Parse a tile 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_mirror_pad

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

Parse a mirror pad 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_scatter_nd

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

Parse a SCATTER_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, subgraph_index: int = 0) -> 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_reverse_v2

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

Parse a REVERSE_V2 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_reverse_sequence

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

Parse a REVERSE_SEQUENCE 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_gru

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

Parse the native CUSTOM(GRU) rolled float operator.

The canonical input order is sequence, initial hidden state, packed input weights, packed recurrent weights, packed input bias, and packed recurrent bias. Outputs are the hidden-state sequence and final hidden state.

parse_unidirectional_sequence_lstm

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

Parse a UNIDIRECTIONAL_SEQUENCE_LSTM operator.

LiteRT fixes this operator to 24 input slots. Optional tensors are represented with -1 and are omitted from named_tensors.

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.

parse_resize_nearest_neighbor

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

Parse a Resize Nearest Neighbor 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_resize_bilinear

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

Parse a Resize Bilinear 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_dilate

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

Parse a Dilate 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_abs

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

Parse an absolute value 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_sqrt

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

Parse a square-root 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_rsqrt

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

Parse a reciprocal square-root 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_where

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

Parse a WHERE operator.