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:1006

AirGruOptions()

Options for the native float16 rolled GRU operator.

attribute

hidden_size

Python

helia_aot/air/options.py:1016

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:1017

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:1018

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

Whether the reset gate is applied after recurrent matmul.