已合并
ascend950新增支持segment sum算子 #1154
zhang-wenbo-beat创建于 1月29日
ascend950新增支持segment sum算子 #1154
已合并
Pull Request已成功合入, 合并人@CANN-robot
（感谢 zhang-wenbo-beat 的贡献）
CANN-robot
1月29日 评论：
1月29日 评论：
CANN-robot
1月29日 评论：
1月29日 评论：
以下是根据您提交的修改文件推荐的Reviewer和Committer序列，需各模块评审通过后方可合入
| Module List | Reviewers | Committers |
|---|---|---|
| index | N/A | chenqi317, liubo75, crystalhu, wangyongguang, zhou-qilong |
| */*/op_graph/*_proto.h | N/A | tang-lei01, wangyongguang |
| */*/op_host/*_def.cpp | N/A | wangyongguang, tang-lei01 |
| scripts | N/A | liubo75, crystalhu, zhou-qilong, fanqirui, tangweiwei2 |
CANN-robot1月29日 将FelixTang7,crystalhu,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
1月29日 将FelixTang7,crystalhu,fanqirui,zhou-qilong,chenqi317,liubo75,tangweiwei2,wangyongguang,zhang-wu设为审查人
1月29日 添加了label：cann-cla/yes
CANN-robot
1月29日 评论：
1月29日 评论：
🔵 source code change are detected, tasks labels is removed in this pull request!
此处折叠了351条消息 查看更多
1月31日 添加了label：lgtm
CANN-robot
1月31日 评论：
1月31日 评论：
Review Guide
This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: chenqi317, zhou-qilong, tang-lei01.
Reviewers who wrote a comment of /lgtm are: chenqi317, zhou-qilong, tang-lei01.
1月31日 关闭了关联的issue
1月31日 合入了pull request
描述
ascend950新增SegmentSum算子，根据segment_ids对输入张量进行分段聚合求和。
关联的Issue
https://gitcode.com/cann/ops-nn/issues/668
测试
1、白盒用例测试，包含各种数据类型的组合，各种分核场景。
2、门槛用例，随机泛化150条用例。
3、网络用例，150条。
4、geir、tf通路验证。
以上测试精度均已通过。特别需要注意，浮点数由于累加序不同，可能会出现非确定性结果。本PR中暂无确定性实现。
文档更新
类型标签