AirUnidirectionalSequenceLstmOptions

AirUnidirectionalSequenceLstmOptions()

Options for the UNIDIRECTIONAL_SEQUENCE_LSTM operator.

Provisional: importable, but outside the compatibility promise; a minor release can change or remove it.

activation: AirActivationType = Field(AirActivationType.NONE, description='Fused activation function')

Fused activation function.

cell_clip: float = Field(0.0, description='Cell-state clipping value')

Cell-state clipping value.

proj_clip: float = Field(0.0, description='Projection clipping value')

Projection clipping value.

time_major: bool = Field(False, description='Whether sequence tensors are time-major')

Whether sequence tensors are time-major.

asymmetric_quantize_inputs: bool = Field(False, description='Whether inputs are asymmetrically quantized')

Whether inputs are asymmetrically quantized.

diagonal_recurrent_tensors: bool = Field(False, description='Whether recurrent tensors use diagonal layout')

Whether recurrent tensors use diagonal layout.

hidden_quantization: AirQuantizationParameters | None = Field(None, description='Quantization of the hidden intermediate (operator intermediate 4)')

Quantization of operator intermediate 4 of an int8 LSTM, which LiteRT’s integer kernel uses to requantize the hidden state. None when the input is not int8, the operator has no intermediate 4, or the intermediate carries no quantization parameters; parameters without a scale are not compared.