slice

Classes

SliceOperator

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

Functions

validate

validate()

Validate the SLICE configuration.

Mirrors :class:StridedSliceOperator because both lower through strided_slice.c.j2 and, on the float path, through the same arm_strided_slice_{f16,f32} kernel.

Raises:

• ConfigValueError –

  If the operator has the wrong tensor count, a rank the codegen cannot express, an unsupported or mismatched dtype, a float16 tensor on a target without FP16 support, or (on the float path) a declared output smaller than the slice extent.

Functions