| slice_with_axes and slice_last_dim for ascend950 Co-authored-by: zhoulong50_hw<zhoulong50@huawei.com> # message auto-generated for no-merge-commit merge: !3315 merge sod-up into master slice_with_axes and slice_last_dim for ascend950 Created-by: zhoulong50_hw Commit-by: zhoulong50_hw Merged-by: cann-robot Description: ## 描述 新增 SliceLastDim 和 SliceWithAxes 两个 conversion 类算子: - SliceLastDim:对输入张量的最后一个维度按 start/end/stride 进行切片,针对连续拷贝和跨步拷贝两种模式做了专门优化 - SliceWithAxes:对输入张量沿指定 axes 按 offsets/size 进行多轴并行切片 包含完整的 proto、infershape、tiling、kernel 实现及 UT。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1964 ## 测试 - 二级冒烟通过 - ttk kernel 算子泛化精度测试:SliceLastDim 100/100 PASS,SliceWithAxes 100/100 PASS - infershape UT:各 5 条,全部 PASS - tiling UT:各 2 条,全部 PASS ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!3315 | 1 个月前 |