| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
nn仓芯片类型补充以及资料格式修改 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !3420 merge xinpian into master nn仓芯片类型补充以及资料格式修改 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3420 | 5 个月前 | |
LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Co-authored-by: ZhouChuping<zhouchuping1@huawei.com> # message auto-generated for no-merge-commit merge: !576 merge master into master LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Created-by: ZhouChuping Commit-by: ZhouChuping Merged-by: cann-robot Description: ## 描述 1.LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 ## 关联的Issue 关联Issue [#316](https://gitcode.com/cann/ops-nn/issues/316) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!576 | 8 个月前 | |
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 | 8 个月前 | |
【新增需求】use npuarch to replace socversion Co-authored-by: 黄强<huangqiang3@huawei.com> # message auto-generated for no-merge-commit merge: !1480 merge master into master 【新增需求】use npuarch to replace socversion Created-by: huang-qiang Commit-by: 黄强 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 优化代码对芯片版本的判断逻辑,使用arch替换芯片名字,提升维护性 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/818 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1480 | 7 个月前 | |
LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Co-authored-by: ZhouChuping<zhouchuping1@huawei.com> # message auto-generated for no-merge-commit merge: !576 merge master into master LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Created-by: ZhouChuping Commit-by: ZhouChuping Merged-by: cann-robot Description: ## 描述 1.LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 ## 关联的Issue 关联Issue [#316](https://gitcode.com/cann/ops-nn/issues/316) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!576 | 8 个月前 | |
修复 scatter_elements 确定性模板ub分配问题,改为2阶段init Co-authored-by: 吕梦鸽<lvmengge@huawei.com> # message auto-generated for no-merge-commit merge: !3092 merge master into master 修复 scatter_elements 确定性模板ub分配问题,改为2阶段init Created-by: lvmengge Commit-by: 吕梦鸽 Merged-by: cann-robot Description: ## 描述 fix scatter_elements determin CopyDataToY 后reset ub ## 关联的Issue 关联Issue [#1687](https://gitcode.com/cann/ops-nn/issues/1687) ## 测试 已进行冒烟测试和st测试 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ √] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3092 | 5 个月前 | |
修复 scatter_elements 确定性模板ub分配问题,改为2阶段init Co-authored-by: 吕梦鸽<lvmengge@huawei.com> # message auto-generated for no-merge-commit merge: !3092 merge master into master 修复 scatter_elements 确定性模板ub分配问题,改为2阶段init Created-by: lvmengge Commit-by: 吕梦鸽 Merged-by: cann-robot Description: ## 描述 fix scatter_elements determin CopyDataToY 后reset ub ## 关联的Issue 关联Issue [#1687](https://gitcode.com/cann/ops-nn/issues/1687) ## 测试 已进行冒烟测试和st测试 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ √] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3092 | 5 个月前 | |
LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Co-authored-by: ZhouChuping<zhouchuping1@huawei.com> # message auto-generated for no-merge-commit merge: !576 merge master into master LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 Created-by: ZhouChuping Commit-by: ZhouChuping Merged-by: cann-robot Description: ## 描述 1.LeakyRelu/LeakyReluGrad/QuantUpdateScatter/ScatterAdd/ScatterElements/ScatterNd/ScatterNdAdd/Swish算子文档补充 ## 关联的Issue 关联Issue [#316](https://gitcode.com/cann/ops-nn/issues/316) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!576 | 8 个月前 | |
CMakeList 中增加算子支持的型号 Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !2143 merge nn_txt into master CMakeList 中增加算子支持的型号 Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> CMakeList 中增加算子支持的型号 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1287 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2143 | 6 个月前 | |
nn仓芯片类型补充以及资料格式修改 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !3420 merge xinpian into master nn仓芯片类型补充以及资料格式修改 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3420 | 5 个月前 |
ScatterElements
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
算子功能:返回输入张量中的唯一元素。
参数说明
-
self(aclTensor*,计算输入):Device侧的aclTensor,数据格式支持ND,维度不大于8。
- Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持BOOL、FLOAT、FLOAT16、DOUBLE、UINT8、INT8、UINT16、INT16、INT32、UINT32、UINT64、INT64、BFLOAT16。
-
sorted(bool,计算输入): 表示是否对 valueOut 按升序进行排序。true时排序,false时乱序。
-
returnInverse(bool,计算输入): 表示是否返回输入数据中各个元素在 valueOut 中的下标。
-
valueOut(aclTensor*,计算输出): Device侧的aclTensor,第一个输出张量,输入张量中的唯一元素,数据格式支持ND。
- Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持BOOL、FLOAT、FLOAT16、DOUBLE、UINT8、INT8、UINT16、INT16、INT32、UINT32、UINT64、INT64、BFLOAT16。
-
inverseOut(aclTensor*,计算输出): 第二个输出张量,当returnInverse为True时有意义,返回self中各元素在valueOut中出现的位置下标,数据类型支持INT64,shape与self保持一致。
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_aclnn_unique | 通过aclnnUnique接口方式调用ScatterElements算子。 |