AirGruOptions

Machine-readable model

• AirGruOptionsclassOptions for the native float16 rolled GRU operator.

class

AirGruOptions

Python

Options for the native float16 rolled GRU operator.

helia_aot/air/options.py:1091

AirGruOptions()

Options for the native float16 rolled GRU operator.

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

attribute

hidden_size

Python

helia_aot/air/options.py:1101

hidden_size: int = Field(..., gt=0, description='Number of hidden units')

Number of recurrent hidden units.

attribute

time_major

Python

helia_aot/air/options.py:1102

time_major: bool = Field(False, description='Use time-major sequence layout')

Whether sequence tensors use [time, batch, feature].

attribute

reset_after

Python

helia_aot/air/options.py:1103

reset_after: bool = Field(True, description='Apply reset after recurrent matmul')

Whether the reset gate is applied after recurrent matmul.