master

cann-robotfeat: 新增DeformableConv2d/LSTM/LSTMCell三个ONNX算子插件

b573ccf1创建于 8 天前历史提交

文件	最后提交记录	最后更新时间
CMakeLists.txt	feat: 新增DeformableConv2d/LSTM/LSTMCell三个ONNX算子插件 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !7581 merge master into master feat: 新增DeformableConv2d/LSTM/LSTMCell三个ONNX算子插件 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述 新增 DeformableConv2d、LSTM、LSTMCell 三个 ONNX 算子的框架插件，用于支持 ONNX 模型到 Ascend GE IR 的算子映射。 ### 新增文件 | 文件 | 说明 | |------|------| | conv/deformable_conv2d/framework/CMakeLists.txt | 可变形卷积2d ONNX 插件构建配置 | | conv/deformable_conv2d/framework/npu_deformable_conv2d_onnx_plugin.cpp | ONNX DeformableConv2d 算子解析与注册 | | rnn/dynamic_rnn/framework/CMakeLists.txt | 动态 RNN ONNX 插件构建配置 | | rnn/dynamic_rnn/framework/npu_lstm_onnx_plugin.cpp | ONNX LSTM 算子解析与注册 | | rnn/dynamic_rnnv2/framework/CMakeLists.txt | 动态 RNNv2 ONNX 插件构建配置 | | rnn/dynamic_rnnv2/framework/npu_lstm_cell_onnx_plugin.cpp | ONNX LSTMCell 算子解析与注册 | ### 实现方式 每个插件通过 REGISTER_CUSTOM_OP 宏注册为 FrameworkType(ONNX) + ImplyType(TVM) 的定制算子，将 ONNX NodeProto 属性解析后映射到 Ascend GE 算子。 ## 测试 - ONNX 模型 DeformableConv2d / LSTM / LSTMCell 算子兼容性验证 ## 类型标签 - [x] 新特性 See merge request: cann/ops-nn!7581	8 天前
npu_deformable_conv2d_onnx_plugin.cpp	feat: 新增DeformableConv2d/LSTM/LSTMCell三个ONNX算子插件 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !7581 merge master into master feat: 新增DeformableConv2d/LSTM/LSTMCell三个ONNX算子插件 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述 新增 DeformableConv2d、LSTM、LSTMCell 三个 ONNX 算子的框架插件，用于支持 ONNX 模型到 Ascend GE IR 的算子映射。 ### 新增文件 | 文件 | 说明 | |------|------| | conv/deformable_conv2d/framework/CMakeLists.txt | 可变形卷积2d ONNX 插件构建配置 | | conv/deformable_conv2d/framework/npu_deformable_conv2d_onnx_plugin.cpp | ONNX DeformableConv2d 算子解析与注册 | | rnn/dynamic_rnn/framework/CMakeLists.txt | 动态 RNN ONNX 插件构建配置 | | rnn/dynamic_rnn/framework/npu_lstm_onnx_plugin.cpp | ONNX LSTM 算子解析与注册 | | rnn/dynamic_rnnv2/framework/CMakeLists.txt | 动态 RNNv2 ONNX 插件构建配置 | | rnn/dynamic_rnnv2/framework/npu_lstm_cell_onnx_plugin.cpp | ONNX LSTMCell 算子解析与注册 | ### 实现方式 每个插件通过 REGISTER_CUSTOM_OP 宏注册为 FrameworkType(ONNX) + ImplyType(TVM) 的定制算子，将 ONNX NodeProto 属性解析后映射到 Ascend GE 算子。 ## 测试 - ONNX 模型 DeformableConv2d / LSTM / LSTMCell 算子兼容性验证 ## 类型标签 - [x] 新特性 See merge request: cann/ops-nn!7581	8 天前