文件最后提交记录最后更新时间
4 天前
1 个月前
5 天前
11 天前
7 个月前
6 天前
4 天前
5 天前
6 个月前
1 个月前
README

FastGelu

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能: 输入fast_gelu的计算结果。
  • 计算公式:

FastGelu(xi)=xi1+exp⁡(−1.702∗∣xi∣)∗exp⁡(0.851∗(xi−∣xi∣))FastGelu(x_i) = \frac {x_i} {1 + \exp(-1.702 * \left| x_i \right|)} * \exp(0.851 * (x_i - \left| x_i \right|))

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入一个张量。 DT_BF16、FLOAT16、DT_FLOAT ND
y 输出 输出一个张量。 DT_BF16、FLOAT16、DT_FLOAT ND

约束说明

调用说明

调用方式 样例代码 说明
aclnn调用 test_aclnn_fast_gelu 通过aclnnFastGelu接口方式调用fast_gelu算子。