已合并
[CANNBOT]ThresholdGradV2D适配Ascend950 Ascend实现 #6179
liuyi2025创建于 6月17日
[CANNBOT]ThresholdGradV2D适配Ascend950 Ascend实现 #6179
已合并
liuyi2025创建于 6月17日
liuyi2025
6月17日

描述

  • 算子功能:完成threshold正向操作的反向传播梯度计算。当 threshold==0 时等价于 ReluGrad。

    • 计算公式:

      outi={gradOutputi,selfi>threshold0,selfithreshold out_i = \begin{cases} gradOutput_i, \quad self_i > threshold\\ 0, \quad self_i \leq threshold \end{cases}

关联的Issue

https://gitcode.com/cann/ops-nn/issues/3387

测试

OBP冒烟:22905
二级冒烟:6637

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 liuyi2025 的贡献)
Lliuyi2025
6月17日 创建了 pull request,commit 49913df7
CANN-robotCANN-robot成员
6月17日 添加了label:stat/needs-squash
CANN-robotCANN-robot成员
6月17日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
6月17日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
*/*/README.md 陈娇, 陈琦 (2/2) 陈娇 (1/1)
*/*/op_api/*.h 汤磊, 陈琦 (2/2) 汤磊 (1/1)
*/*/op_host/*_def.cpp 汤磊, 陈琦 (2/2) 汤磊 (1/1)
activation 黄迪, 陈琦, 王星 (3/2) 陈琦, 王星, 黄迪 (3/1)
docs 陈琦, 王星 (2/2) 陈琦, 王星 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

liuyi2025, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
liuyi2025
6月17日 评论:

compile

likedislike
此处折叠了87条消息 查看更多
CANN-robot
CANN-robot成员
6月18日 评论:

The MR can not be merged, because of CodeReview discussion not resolved

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike
NerdddddddddddNerddddddddddd成员
6月18日 解决了最后一个问题
CANN-robotCANN-robot成员
6月18日 关闭了关联的issue
CANN-robotCANN-robot成员
6月18日 合入了pull request
CANN-robot
CANN-robot成员
6月18日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike