gru

AOT lowering for the native rolled float16 GRU operator.

Classes

GruOperator

GruOperator(op: AirOperator, model: AirModel, platform: SocPlatform, prefix: str = 'aot', attributes: dict[str, str] | None = None)

Lower CUSTOM(GRU) to the NS-CMSIS-NN rolled float16 kernel.

Create a rolled GRU operator.

Functions

validate

validate() -> None

Validate the canonical stateful float16 GRU contract.

on_resolve

on_resolve() -> None

Validate GRU configuration during operator resolution.

compute_values

compute_values() -> dict[str, Any]

Return values used by the GRU C templates.

emit

emit(save_path: Path) -> None

Emit the GRU operator source and header.

Functions