gather

Classes

GatherOperator

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

GATHER operator.

This operator gathers slices from params along the specified axis.

Parameters:

• op

  (AirOperator) –

  The AIR operator to wrap.

• model

  (AirModel) –

  The AIR model.

• platform

  (SocPlatform) –

  Target platform for code generation.

• prefix

  (str, default: 'aot' ) –

  Prefix for generated code files. Defaults to "aot".

• attributes

  (dict[str, str] | None, default: None ) –

  Attributes for template values.

Functions

Functions