文件最后提交记录最后更新时间
feat: 新增 DynamicStitch 算子实现 Co-authored-by: 王荀<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !1633 merge dynamic_stitch_fix into master feat: 新增 DynamicStitch 算子实现 Created-by: wangxun21 Commit-by: wangxun21;王荀 Merged-by: cann-robot Description: ## 描述 本 PR 新增了 DynamicStitch 算子的完整实现,该算子与 TensorFlow 的 DynamicStitch 算子兼容。 DynamicStitch 算子用于将多个张量的数据交织到单个张量中,支持多种数据类型。 ### 改动原因 支持 TensorFlow 模型中 DynamicStitch 算子的迁移,丰富 ops-math 仓库的算子支持范围。 ### 改动方法 1. 新增算子原型定义(op_graph/dynamic_stitch_proto.h) 2. 新增 host 侧实现: - 算子属性配置(op_host/dynamic_stitch_def.cpp) - Tiling 配置(op_host/config/ascend950/) 3. 新增 kernel 侧实现: - 通用头文件(op_kernel/arch35/dynamic_stitch_common.h) - Indices 去重(op_kernel/arch35/dynamic_stitch_indices_deduplicate.h) - SIMD/SIMT 实现(op_kernel/arch35/) 4. 新增单元测试(tests/ut/) 5. 新增示例代码(examples/test_geir_dynamic_stitch.cpp) ## 关联的Issue - #998 ## 测试 - 已添加单元测试 test_dynamic_stitch_tiling.cpp - 已添加示例代码 test_geir_dynamic_stitch.cpp ## 文档更新 无 ## 类型标签 - [x] 新特性 See merge request: cann/ops-math!16332 个月前
feat: 新增 DynamicStitch 算子实现 Co-authored-by: 王荀<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !1633 merge dynamic_stitch_fix into master feat: 新增 DynamicStitch 算子实现 Created-by: wangxun21 Commit-by: wangxun21;王荀 Merged-by: cann-robot Description: ## 描述 本 PR 新增了 DynamicStitch 算子的完整实现,该算子与 TensorFlow 的 DynamicStitch 算子兼容。 DynamicStitch 算子用于将多个张量的数据交织到单个张量中,支持多种数据类型。 ### 改动原因 支持 TensorFlow 模型中 DynamicStitch 算子的迁移,丰富 ops-math 仓库的算子支持范围。 ### 改动方法 1. 新增算子原型定义(op_graph/dynamic_stitch_proto.h) 2. 新增 host 侧实现: - 算子属性配置(op_host/dynamic_stitch_def.cpp) - Tiling 配置(op_host/config/ascend950/) 3. 新增 kernel 侧实现: - 通用头文件(op_kernel/arch35/dynamic_stitch_common.h) - Indices 去重(op_kernel/arch35/dynamic_stitch_indices_deduplicate.h) - SIMD/SIMT 实现(op_kernel/arch35/) 4. 新增单元测试(tests/ut/) 5. 新增示例代码(examples/test_geir_dynamic_stitch.cpp) ## 关联的Issue - #998 ## 测试 - 已添加单元测试 test_dynamic_stitch_tiling.cpp - 已添加示例代码 test_geir_dynamic_stitch.cpp ## 文档更新 无 ## 类型标签 - [x] 新特性 See merge request: cann/ops-math!16332 个月前