AirActivationType

AirActivationType()

AIR activation function types.

NONE: str = auto()

No activation.

RELU: str = auto()

ReLU activation.

RELU_N1_TO_1: str = auto()

ReLU activation with range [-1, 1].

RELU_0_TO_1: str = auto()

ReLU activation with range [0, 1].

RELU6: str = auto()

ReLU activation with range [0, 6].

TANH: str = auto()

Tanh activation.

SIGN_BIT: str = auto()

Sign bit activation.

PRELU: str = auto()

Parametric ReLU activation.

compute_range(
    output_zero_point: int,
    output_scale: float,
    *,
    dtype: npt.DTypeLike | None = None,
    qmin: int | None = None,
    qmax: int | None = None,
    num_bits: int | None = None,
    signed: bool | None = None,
) -> tuple[int, int]

Compute quantized activation clamp range.