{
  "$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": "AirStridedSliceOptions",
      "path": "helia_aot.air.options.AirStridedSliceOptions",
      "submodules": [],
      "summary": "",
      "symbols": [
        {
          "description": "Options for STRIDED_SLICE operator.",
          "examples": [],
          "id": "helia_aot.air.options.AirStridedSliceOptions",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "Begin indices for each dimension.",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.begin",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "begin",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "begin: tuple[int, int, int, int] = Field(..., description='Begin indices for each dimension')",
              "source": {
                "line": 794,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L794"
              },
              "summary": ""
            },
            {
              "description": "End indices for each dimension.",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.end",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "end",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "end: tuple[int, int, int, int] = Field(..., description='End indices for each dimension')",
              "source": {
                "line": 795,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L795"
              },
              "summary": ""
            },
            {
              "description": "Stride values for each dimension.",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.stride",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "stride",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "stride: tuple[int, int, int, int] = Field(..., description='Stride values for each dimension')",
              "source": {
                "line": 796,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L796"
              },
              "summary": ""
            },
            {
              "description": "Mask for begin indices (default 0).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.begin_mask",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "begin_mask",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "begin_mask: int = Field(0, description='Mask for begin indices')",
              "source": {
                "line": 798,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L798"
              },
              "summary": ""
            },
            {
              "description": "Mask for end indices (default 0).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.end_mask",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "end_mask",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "end_mask: int = Field(0, description='Mask for end indices')",
              "source": {
                "line": 799,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L799"
              },
              "summary": ""
            },
            {
              "description": "Mask for ellipsis (default 0).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.ellipsis_mask",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "ellipsis_mask",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "ellipsis_mask: int = Field(0, description='Mask for ellipsis')",
              "source": {
                "line": 800,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L800"
              },
              "summary": ""
            },
            {
              "description": "Mask for new axes (default 0).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.new_axis_mask",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "new_axis_mask",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "new_axis_mask: int = Field(0, description='Mask for new axes')",
              "source": {
                "line": 801,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L801"
              },
              "summary": ""
            },
            {
              "description": "Mask for shrink axes (default 0).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.shrink_axis_mask",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "shrink_axis_mask",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "shrink_axis_mask: int = Field(0, description='Mask for shrink axes')",
              "source": {
                "line": 802,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L802"
              },
              "summary": ""
            },
            {
              "description": "If True, apply offset to the begin and end indices (default False).",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.offset",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "offset",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "offset: bool = Field(False, description='If True, apply offset to the begin and end indices')",
              "source": {
                "line": 803,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L803"
              },
              "summary": ""
            },
            {
              "description": "Compute (begin, end, stride) for each axis, padding dims and shape to `pad_to` dims.",
              "examples": [],
              "id": "helia_aot.air.options.AirStridedSliceOptions.get_slices",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "get_slices",
              "params": [
                {
                  "description": "The shape of the input tensor.",
                  "name": "input_shape",
                  "type": "tuple[int, ...]"
                },
                {
                  "default": "4",
                  "description": "The number of dimensions to pad the input shape to (default 4).",
                  "name": "pad_to",
                  "type": "int"
                }
              ],
              "raises": [],
              "returns": [
                {
                  "description": "A list of AirStrideSlice, each containing (begin, end, stride) for each axis.",
                  "type": "list[AirStrideSlice]"
                }
              ],
              "signature": "get_slices(input_shape: tuple[int, ...], pad_to: int = 4) -> list[AirStrideSlice]",
              "source": {
                "line": 805,
                "path": "helia_aot/air/options.py",
                "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L805"
              },
              "summary": "Compute (begin, end, stride) for each axis, padding dims and shape to padto dims."
            }
          ],
          "name": "AirStridedSliceOptions",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "AirStridedSliceOptions()",
          "source": {
            "line": 778,
            "path": "helia_aot/air/options.py",
            "url": "https://github.com/AmbiqAI/helia-aot/blob/cf2246a7ad439fa38d57462118e7840e5497a685/helia_aot/air/options.py#L778"
          },
          "summary": "Options for STRIDEDSLICE operator."
        }
      ]
    }
  ],
  "name": "helia_aot"
}