{
  "$schema": "https://ambiqai.github.io/helia-ui/schema/reference-model-1.json",
  "generatedFrom": {
    "sourceCommit": "cf2246a7ad439fa38d57462118e7840e5497a685",
    "tool": "pyref",
    "version": "1.7.3"
  },
  "language": "python",
  "modules": [
    {
      "description": "",
      "name": "AirGruOptions",
      "path": "helia_aot.air.options.AirGruOptions",
      "submodules": [],
      "summary": "",
      "symbols": [
        {
          "description": "Options for the native float16 rolled GRU operator.",
          "examples": [],
          "id": "helia_aot.air.options.AirGruOptions",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "Number of recurrent hidden units.",
              "examples": [],
              "id": "helia_aot.air.options.AirGruOptions.hidden_size",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "hidden_size",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "hidden_size: int = Field(..., gt=0, description='Number of hidden units')",
              "source": {
                "line": 1016,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L1016"
              },
              "summary": ""
            },
            {
              "description": "Whether sequence tensors use ``[time, batch, feature]``.",
              "examples": [],
              "id": "helia_aot.air.options.AirGruOptions.time_major",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "time_major",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "time_major: bool = Field(False, description='Use time-major sequence layout')",
              "source": {
                "line": 1017,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L1017"
              },
              "summary": ""
            },
            {
              "description": "Whether the reset gate is applied after recurrent matmul.",
              "examples": [],
              "id": "helia_aot.air.options.AirGruOptions.reset_after",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "reset_after",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "reset_after: bool = Field(True, description='Apply reset after recurrent matmul')",
              "source": {
                "line": 1018,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L1018"
              },
              "summary": ""
            }
          ],
          "name": "AirGruOptions",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "AirGruOptions()",
          "source": {
            "line": 1006,
            "path": "helia_aot/air/options.py",
            "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L1006"
          },
          "summary": "Options for the native float16 rolled GRU operator."
        }
      ]
    }
  ],
  "name": "helia_aot"
}