| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
支持scatter_nd_update功能实现 Co-authored-by: yuhwang<wangyuhang70@huawei.com> # message auto-generated for no-merge-commit merge: !2890 merge scatter_nn into master 支持scatter_nd_update功能实现 Created-by: yuhwang Commit-by: yuhwang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> - 支持scatter_nd_update算子功能实现。 - scatter_nd_update根据indices在给定变量内,将updates应用于变量的单个值或切片。 - ref[indices[k]]=updates[k] - ref是一个维度为 P的张量Tensor; - indices是一个维度为 Q的整型张量Tensor; - indices的shape一定是[d0,..., d_Q-2, K],此处0<K<=P; - updates 是一个维度为Q−1+P−K 的张量:[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]] - 更新op_list.md文档, 补齐avg_pool, avg_pool_v2在op_list.md的链接,修改avg_pool readme中的一处用词选择。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1616 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ttk测试,衡芯st测试,opp冒烟测试,ut测试,单算子aclnn example测试 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> - 更新了/index/scatter_nd_update/READ.md 文件 - 更新了/pooling/avg_pool/READ.md 文件 - 更新了/docs/zh/op_list.md 文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!2890 | 4 个月前 | |
scatter_nd_update算子移除float8_e8m0算子类型 Co-authored-by: yuhwang<wangyuhang70@huawei.com> # message auto-generated for no-merge-commit merge: !3448 merge scatter_nn into master scatter_nd_update算子移除float8_e8m0算子类型 Created-by: yuhwang Commit-by: yuhwang Merged-by: cann-robot Description: ## 描述 走GEIR时,对于部分算子GE会在前面插入一个TensorMove算子,部分算子(transpose,tensormove,as_stride,viewcopy,slice,broadcastto)不支持float8_e8m0, 导致ScatterNdUpdate该类型的图模式无法打通,故scatter_nd_update算子删除float8_e8m0 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1897 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 1. 衡芯st 2.opp冒烟测试 3. xrun geir测试 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> index/scatter_nd_update/README.md index/scatter_nd_update/docs/aclnnScatterNdUpdate.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3448 | 4 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 4 个月前 | ||
| 4 个月前 |