Star0

Star0

cann-robot删除源码链接

898b294cAdd on 24 days agoHistory

Files	Last commit	Last update
docs	删除源码链接 Co-authored-by: fwx1169505<fengying27@h-partners.com> # message auto-generated for no-merge-commit merge: !12106 merge master into master 删除源码链接 Created-by: fengying555682 Commit-by: fwx1169505 Merged-by: cann-robot Description: ## 描述删除源码链接 ## 关联的Issue [#6769](https://gitcode.com/cann/ops-nn/issues/6769) ## 测试仅涉及文档修改，已经测试md文档语法 ## 文档更新 activation和quant目录下的md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!12106	24 days ago
examples	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
framework	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
op_api	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
op_graph	legacy下线：整改NN算子图模式注册与日志规范 Co-authored-by: zhouxuan78<zhouxuan78@huawei.com> # message auto-generated for no-merge-commit merge: !8388 merge master into master legacy下线：整改NN算子图模式注册与日志规范 Created-by: zhouxuan78 Commit-by: zhouxuan78 Merged-by: cann-robot Description: ## 描述本 PR 用于 ops-nn 仓 Legacy 下线整改，补齐部分算子的直调/图模式注册、InferShape / InferDataType 拆分、日志规范整改及必要的编译/预提交问题修复。 1. 补齐多个算子的 `op_graph` 侧 InferDataType / graph infer 注册文件，并同步补充对应 `CMakeLists.txt`： - 如 FastGeluV2、HardShrinkGrad、HardSigmoidGrad、HardSwish、HardSwishGrad、Selu、SeluGrad、Shrink、SoftplusGrad、SoftplusV2、SoftplusV2Grad、Softsign、SoftsignGrad、SmoothL1Loss、SoftMarginLoss、LpNormUpdateV2、ApplyFtrlV2、ApplyMomentum、ApplyRMSProp 等。 2. 调整部分算子的 `op_host` infer 文件： - 将 InferDataType 逻辑从 infershape 文件中拆分/迁移到 `op_graph` 侧； - 保留 InferShape 逻辑，避免 shape/type 混在同一处注册； - 收敛 IndexCheck infershape 日志，仅保留入口日志。 3. 补充 SoftMarginLoss 图模式原型： - 新增 `loss/soft_margin_loss/op_graph/soft_margin_loss_proto.h`； - 新增 `OPS_PROTO_DEF_SOFTMARGINLOSS` 去重宏，避免重复注册； - 补充 `SoftMarginLoss` 的 `op_graph/CMakeLists.txt` 和 graph infer 注册。 4. 日志整改： - 为 Arch35 tiling 入口补充统一入口日志； - 对本 PR 修改范围内新增/调整的外部输入错误日志，按规范使用 `OP_LOGE_FOR_INVALID_*` 系列上报接口； - 收敛部分过多的 tiling / infershape 日志打印，避免循环或冗余打印。 5. 回退/收敛不属于本次整改范围的 Softshrink 重命名相关改动： - Softshrink -> SoftShrink 的命名调整不放在本 PR 中，后续单独 PR 处理。 6. 修复 pre-commit 格式问题： - 按仓库 `.clang-format` 修复格式； - 不修改 `.pre-commit-config.yaml`。影响范围主要影响 activation、index、loss、norm、optim、quant、vfusion 等目录下 Legacy 下线相关算子的 op_graph 注册、InferDataType 拆分、tiling 入口日志和错误日志规范。 ## 变更内容 ## 关联的Issue ## 测试 ## 文档更新 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：Legacy包下线整改 See merge request: cann/ops-nn!8388	2 months ago
op_host	legacy下线：整改NN算子图模式注册与日志规范 Co-authored-by: zhouxuan78<zhouxuan78@huawei.com> # message auto-generated for no-merge-commit merge: !8388 merge master into master legacy下线：整改NN算子图模式注册与日志规范 Created-by: zhouxuan78 Commit-by: zhouxuan78 Merged-by: cann-robot Description: ## 描述本 PR 用于 ops-nn 仓 Legacy 下线整改，补齐部分算子的直调/图模式注册、InferShape / InferDataType 拆分、日志规范整改及必要的编译/预提交问题修复。 1. 补齐多个算子的 `op_graph` 侧 InferDataType / graph infer 注册文件，并同步补充对应 `CMakeLists.txt`： - 如 FastGeluV2、HardShrinkGrad、HardSigmoidGrad、HardSwish、HardSwishGrad、Selu、SeluGrad、Shrink、SoftplusGrad、SoftplusV2、SoftplusV2Grad、Softsign、SoftsignGrad、SmoothL1Loss、SoftMarginLoss、LpNormUpdateV2、ApplyFtrlV2、ApplyMomentum、ApplyRMSProp 等。 2. 调整部分算子的 `op_host` infer 文件： - 将 InferDataType 逻辑从 infershape 文件中拆分/迁移到 `op_graph` 侧； - 保留 InferShape 逻辑，避免 shape/type 混在同一处注册； - 收敛 IndexCheck infershape 日志，仅保留入口日志。 3. 补充 SoftMarginLoss 图模式原型： - 新增 `loss/soft_margin_loss/op_graph/soft_margin_loss_proto.h`； - 新增 `OPS_PROTO_DEF_SOFTMARGINLOSS` 去重宏，避免重复注册； - 补充 `SoftMarginLoss` 的 `op_graph/CMakeLists.txt` 和 graph infer 注册。 4. 日志整改： - 为 Arch35 tiling 入口补充统一入口日志； - 对本 PR 修改范围内新增/调整的外部输入错误日志，按规范使用 `OP_LOGE_FOR_INVALID_*` 系列上报接口； - 收敛部分过多的 tiling / infershape 日志打印，避免循环或冗余打印。 5. 回退/收敛不属于本次整改范围的 Softshrink 重命名相关改动： - Softshrink -> SoftShrink 的命名调整不放在本 PR 中，后续单独 PR 处理。 6. 修复 pre-commit 格式问题： - 按仓库 `.clang-format` 修复格式； - 不修改 `.pre-commit-config.yaml`。影响范围主要影响 activation、index、loss、norm、optim、quant、vfusion 等目录下 Legacy 下线相关算子的 op_graph 注册、InferDataType 拆分、tiling 入口日志和错误日志规范。 ## 变更内容 ## 关联的Issue ## 测试 ## 文档更新 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他：Legacy包下线整改 See merge request: cann/ops-nn!8388	2 months ago
op_kernel	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
tests	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	4 months ago
CMakeLists.txt	[CANNBot]新增Shrink算子AscendC适配Ascend950 Co-authored-by: Coder_Nerd<shishuai5@huawei.com> # message auto-generated for no-merge-commit merge: !5184 merge operators/migrate-shrink-builtin into master [CANNBot]新增Shrink算子AscendC适配Ascend950 Created-by: Coder_Nerd Commit-by: Coder_Nerd Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 新增Shrink算子AscendC适配Ascend950 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2851 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 二级冒烟 5658 通过 OBP冒烟 22282 通过使用ops-test-kit执行549条用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5184	5 months ago
README.md	修改产品名 Co-authored-by: fwx1169505<fengying27@h-partners.com> # message auto-generated for no-merge-commit merge: !11017 merge master into master 修改产品名 Created-by: fengying555682 Commit-by: fwx1169505 Merged-by: cann-robot Description: ## 描述修改quant和activation文件夹中的md文档涉及的产品名称 ## 关联的Issue [#6056](https://gitcode.com/cann/ops-nn/issues/6056) ## 测试已经完成文档语法和链接一致性检测 ## 文档更新 quant和activation文件夹中的md文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11017	1 month ago

Shrink

产品支持情况

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

功能说明

算子功能：对输入张量进行非线性收缩变换，根据输入值与阈值lambd的关系将元素映射到三个区域（正区域、负区域、死区），常用于稀疏化模型推理和训练中的权重正则化。
计算公式：
$outi={selfi−bias,selfi>lambdselfi+bias,selfi<−lambd0,−lambd≤selfi≤lambdout_i=\begin{cases} self_i-bias, & self_i > lambd \\ self_i+bias, & self_i < -lambd \\ 0, & -lambd \leq self_i \leq lambd \end{cases}$
其中lambd为阈值参数（默认0.5），bias为偏移量参数（默认0.0）。当lambd < 0时，内部钳位为0。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input_x	输入	输入张量，对应公式中的self。元素值与lambd比较决定归属区域。	FLOAT16、FLOAT	ND
lambd	属性	阈值参数，对应公式中的lambd。当self_i>lambd时进入正区域，self_i<-lambd时进入负区域。默认值为0.5。当lambd<0时，内部钳位为0。	FLOAT	-
bias	属性	偏移量参数，对应公式中的bias。在正区域减去bias，在负区域加上bias。默认值为0.0。	FLOAT	-
output_y	输出	输出张量，对应公式中的out。与input_x的dtype和shape完全相同。	FLOAT16、FLOAT	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_shrink	通过aclnnShrink接口方式调用Shrink算子。
图模式	-	通过算子IR构图方式调用Shrink算子。