文件最后提交记录最后更新时间
14 天前
1 个月前
1 个月前
1 个月前
1 个月前
6 个月前
2 个月前
README

SwishGrad

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品

功能说明

  • 算子功能:求swish函数梯度。

  • 计算公式:

y=sigmoid(scale∗x)+x∗sigmoid(scale∗x)y = sigmoid(scale*x) + x*sigmoid(scale*x)

sigmoid=sigmoid∗(1−sigmoid)sigmoid = sigmoid*(1 - sigmoid)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
grad 输入 待进行SwishGrad计算的入参,公式中的grad。 fp16、fp32、bf16 ND,FRACTAL_NZ,NC1HWC0
x 输入 待进行SwishGrad计算的入参,公式中的x。 fp16、fp32、bf16 ND,FRACTAL_NZ,NC1HWC0
scale 输入 待进行SwishGrad计算的入参,公式中的scale。 fp32 1
y 输入 待进行SwishGrad计算的入参, 暂时不参与运算。 fp16、fp32、bf16 ND,FRACTAL_NZ,NC1HWC0
grad_x 输出 待进行SwishGrad计算的出参,公式中的输出。 fp16、fp32、bf16 ND,FRACTAL_NZ,NC1HWC0

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_swish_grad.cpp 通过test_aclnn_swish_grad接口方式调用SwishGrad算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
ilovescrapy 个人开发者 SwishGrad 2025/12/23 SwishGrad算子适配开源仓