ActivationType

说明

激活函数类型枚举。

定义

enum ActivationType : int {
    ACTIVATION_UNDEFINED = 0,
    ACTIVATION_RELU,
    ACTIVATION_GELU,
    ACTIVATION_FAST_GELU,
    ACTIVATION_SWISH,
    ACTIVATION_LOG,
    ACTIVATION_SWIGLU_FORWARD,
    ACTIVATION_SWIGLU_BACKWARD,
    ACTIVATION_MAX,
};

成员

成员名称	描述
ACTIVATION_UNDEFINED	未定义。
ACTIVATION_RELU	RELU激活类型。
ACTIVATION_GELU	GELU激活类型。其中bfloat16类型数据仅支持Atlas 800I A2推理产品。
ACTIVATION_FAST_GELU	FAST_GELU激活类型。快速运算的Gelu激活函数，对Tensor内每个element做Gelu激活函数近似计算，计算速度更快，同时保持较高的准确性。
ACTIVATION_SWISH	SWISH激活类型。
ACTIVATION_LOG	LOG激活类型。
ACTIVATION_SWIGLU_FORWARD	SWIGLU_FORWARD激活类型。 Swiglu正向激活函数。对于Atlas 推理系列产品，仅支持32位对齐的数据，不支持bfloat16类型数据。
ACTIVATION_SWIGLU_BACKWARD	SWIGLU_BACKWARD激活类型。 Swiglu正向激活函数的反向，求梯度时使用。当前仅支持Atlas 800I A2推理产品。
ACTIVATION_MAX	枚举最大值, 非激活类型。

父主题： atb/infer_op_params.h