Star0
0
代码
代码
Issues
Pull Requests
Actions
项目讨论
Wiki
分析
Star0
0
cann-robotcann-robotModifying the Chips Supported by an Operator
4251fe00创建于 6月22日历史提交
文件最后提交记录最后更新时间
docs
[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
4 个月前
examples
[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
4 个月前
framework
nn类onnx算子插件支持 Co-authored-by: x00842564<xuyanke@huawei.com> Co-authored-by: ouyangxiantian<ouyangxiantian@h-partners.com> # message auto-generated for no-merge-commit merge: !452 merge master into master nn类onnx算子插件支持 Created-by: yanke-xu Commit-by: x00842564;ouyangxiantian Merged-by: cann-robot Description: ## 描述 涉及的onnx插件算子列表： 1. Elu 2. NPUFastGelu 3. NPUGeglu 4. NPUGeluV2 5. HardSigmoid 6. LeakyRelu 7. Mish 8. Selu 9. Shrink 10. NPUSoftmaxCrossEntropyWithLogits 11. Softmax 12. ThresholdedRelu 13. AdaCast 14. AdaptiveAvgPool2d 15. AdaptiveMaxPool2d 16. AscendAntiQuant 17. AscendDequant 18. AscendDequantS16 19. AscendQuant 20. AscendRequant 21. AscendRequantS16 22. BasicGRUInplaceFillWindowCache 23. BasicLSTMInplaceFillWindowCache 24. BatchMatMul 25. BatchNormalization 26. Celu 27. ConstantOfShape 28. DynSeqOuter 29. GlobalLpPool 30. GlobalMaxPool 31. GroupNormRelu 32. GRU 33. Hardmax 34. If 35. InstanceNormalization 36. Int8Dequantize 37. Int8GivenIntTensorFill 38. Int8Quantize 39. Int8Transpose 40. LogSoftmax 41. LRN 42. LSTM 43. LSTMP 44. MatMul 45. MeanVarianceNormalization 46. MovingSumWithSigmoid 47. NPUAnchorResponseFlags 48. NPUDropoutWithAddSoftmax 49. NPUGru 50. Resize 51. Size 52. SoftmaxCrossEntropyLoss 53. TopK 54. Unique 55. Shape 56. GatherElements 57. GatherND 58. Compress 59. Gather 60. NonZero 61. NPUScatter 62. ScatterElements 63. Scatter 64. QuantBatchMatMul 65. WeightQuantBatchMatmulV2 66. NPUAddRmsNorm 67. NPUGroupNormSilu 68. LpNormalization 69. NPURmsNorm 70. NPUAntiQuant 71. NPUAscendQuantV2 72. NPUDynamicQuant 73. NPUDynamicQuantV2 74. NPUGroupQuant 75. NPUQuantize ## 关联的Issue None ## 测试 编译构建、冒烟测试、静态扫描 ## 文档更新 None ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!452
9 个月前
op_api
[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
4 个月前
op_graph
[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
4 个月前
op_host
[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
4 个月前
op_kernel
[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
4 个月前
tests
[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
4 个月前
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
4 个月前
README.md
Modifying the Chips Supported by an Operator Co-authored-by: zhouwenfang<zhouwenfang3@h-partners.com> # message auto-generated for no-merge-commit merge: !6365 merge master into master Modifying the Chips Supported by an Operator Created-by: zhouwenfang Commit-by: zhouwenfang Merged-by: cann-robot Description: ## 描述 Modifying the Chips Supported by an Operator ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3483 ## 测试 NA ## 文档更新 具体请见issue ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6365
3 个月前
README

Shrink

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT √
Atlas A3 训练系列产品/Atlas A3 推理系列产品 √
Atlas A2 训练系列产品/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算子。