ops-nn/index/scatter_update/op_api · CANN/ops-nn

文件	最后提交记录	最后更新时间
aclnn_index_copy.cpp	C++ clean Code规范 Co-authored-by: doufloat<baijinxiang@huawei.com> # message auto-generated for no-merge-commit merge: !6666 merge cherry-pick-clean-code-to-910 into 9.1.0 C++ clean Code规范 Created-by: doufloat Commit-by: doufloat Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 解决如下Clean code问题：静态扫描出现编译告警： comparison between signed and unsigned integer expressions [-Wsign-compare] type qualifiers ignored on function return type [-Wignored-qualifiers] unused variable narrowing conversion of '(int64_t)bp_groups' from 'int64_t {aka long int}' to 'int' inside { } [-Wnarrowing] C++代码规范： G.FUN.02-CPP 函数的所有声明必须与定义具有一致的参数名 G.FUN.03-CPP 禁止函数有未被使用的参数 G.FMT.17-CPP 合理安排空行 G.EXP.22-CPP 确保除法和余数运算不会导致除零错误(被零除) G.CNS.02-CPP 不要使用难以理解的字面常量 G.CNS.04-CPP 对于指针和引用类型的参数，如果不需要修改其引用的对象，应使用const修饰 G.FMT.14-CPP 指针类型"*"和引用类型"&"只跟随类型或变量名 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3309 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!6666	1 个月前
aclnn_index_copy.h	sorted_sparse_segment_mean_grad和scatter_update算子开源到nn仓 Co-authored-by: 袁晓东<yuanxiaodong6@huawei.com> # message auto-generated for no-merge-commit merge: !2224 merge sortedsparseqianyi into master sorted_sparse_segment_mean_grad和scatter_update算子开源到nn仓 Created-by: xiaodong666 Commit-by: 袁晓东 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> sorted_sparse_segment_mean_grad和scatter_update算子开源到nn仓； scatter_update：将index张量中元素值作为索引，针对指定轴0，把source中元素复制到out的对应位置上。 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出/属性</th> <th>描述</th> <th>使用说明</th> <th>数据类型</th> <th>数据格式</th> <th>维度(shape)</th> </tr></thead> <tbody> <tr> <td>var</td> <td>输入</td> <td>输入Tensor</td> <td>-</td> <td>FLOAT、BFLOAT16、FLOAT16、INT32、UINT32、INT64、UINT64、INT16、INT8、UINT8、DOUBLE、BOOL、COMPLEX128、COMPLEX64</td> <td>ND</td> <td>-</td> </tr> <tr> <td>indices</td> <td>输入</td> <td>输入Tensor</td> <td><ul><li>维度不能大于1，元素个数要求和张量source中dim轴的大小相同。</li><li>当index有重复索引值的时候，结果不保序。</li></ul></td> <td>INT32, INT64</td> <td>ND</td> <td>-</td> </tr> <tr> <td>updates</td> <td>输入</td> <td>输入Tensor</td> <td>dim轴的大小和index的元素数量相同，其他维度同selfRef的shape。</td> <td>FLOAT、BFLOAT16、FLOAT16、INT32、UINT32、INT64、UINT64、INT16、INT8、UINT8、DOUBLE、BOOL、COMPLEX128、COMPLEX64</td> <td>ND</td> <td>-</td> </tr> <tr> <td>var</td> <td>输出</td> <td>输出Tensor</td> <td>-</td> <td>FLOAT、BFLOAT16、FLOAT16、INT32、UINT32、INT64、UINT64、INT16、INT8、UINT8、DOUBLE、BOOL、COMPLEX128、COMPLEX64</td> <td>ND</td> <td>-</td> </tr> </tbody></table> sorted_sparse_segment_mean_grad：计算张量稀疏片段的均值的反向梯度算子 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>x</td> <td>输入</td> <td>输入aclTensor</td> <td>FLOAT, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> <tr> <td>sorted_indices</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>pre_location_indices</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>segment_ids</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>output_dim0</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出aclTensor</td> <td>FLOAT, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> </tbody></table> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> Issue [#1248](https://gitcode.com/cann/ops-nn/issues/1248) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 批跑st用例，geir、tf、torch端到端通路验证，冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2224	4 个月前
scatter_update.cpp	l0头文件迁移 Co-authored-by: hid54649903<zhangjunming21@huawei.com> # message auto-generated for no-merge-commit merge: !5124 merge l0_mv into master l0头文件迁移 Created-by: hid54649903 Commit-by: hid54649903 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 迁移部分l0头文件至common/inc/opapi ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> Issue [#2858](https://gitcode.com/cann/ops-nn/issues/2858) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 流水线通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：开源仓整改 See merge request: cann/ops-nn!5124	2 个月前