master

cann-robotformat cpp

5db91519Add on Jul 6History

Files	Last commit	Last update
fusion_pass	gather_nd算子支持Ascend950 Co-authored-by: ChenPengyi<chenpengyi@huawei.com> # message auto-generated for no-merge-commit merge: !1271 merge cpy_gather_nd1 into master gather_nd算子支持Ascend950 Created-by: cpy_123456 Commit-by: ChenPengyi Merged-by: cann-robot Description: ## 描述 gather_nd算子支持Ascend950（开源迁移） ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/820 算子功能：对于维度为**r≥1**的输入张量self，和维度**q≥1**的输入张量indices，将数据切片收集到维度为 **(q-1) + (r - indices_shape[-1])** 的输出张量out中。 关联pr:https://gitcode.com/cann/canndev/pull/80634 输入输出设计： <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 183px"> <col style="width: 120px"> <col style="width: 265px"> <col style="width: 299px"> <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>self</td> <td>输入</td> <td>输入aclTensor。</td> <td>INT64、INT32、INT8、UINT8、BOOL、FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT16、UINT16、UINT32、UINT64</td> <td>ND</td> </tr> <tr> <td>indices</td> <td>输入</td> <td>输入aclTensor。</td> <td>INT32、INT64</td> <td>ND</td> </tr> <tr> <td>negativeIndexSupport</td> <td>输入</td> <td>onnx模型是否存在负索引。</td> <td>BOOL</td> <td>-</td> </tr> <tr> <td>out</td> <td>输出</td> <td>输出aclTensor。</td> <td>与self一致</td> <td>ND</td> </tr> </tbody></table> ## 测试 通过二级冒烟&david_st测试 ## 文档更新 https://gitcode.com/cann/ops-nn/blob/0a109f44acb5adc6c9d979b78ea516eaea165927/index/gather_nd/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1271	8 months ago
gather_nd_proto.h	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