| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[CANNBot]Experimental 新增Shrink子AscendC适配 Ascend910B/Ascend950 Co-authored-by: Coder_Nerd<shishuai5@huawei.com> # message auto-generated for no-merge-commit merge: !3898 merge ai_gen_longtail into master [CANNBot]Experimental 新增Shrink子AscendC适配 Ascend910B/Ascend950 Created-by: Coder_Nerd Commit-by: Coder_Nerd Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 对输入张量进行非线性收缩变换。根据输入值与阈值 lambd 的大小关系,通过偏移量 bias 进行缩放和偏移处理。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2068 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> shrink/ ├── CMakeLists.txt ├── README.md ├── examples/ │ └── test_aclnn_shrink.cpp ├── op_host/ │ ├── CMakeLists.txt │ ├── shrink_def.cpp │ ├── shrink_infershape.cpp │ └── shrink_tiling.cpp ├── op_kernel/ │ ├── shrink.cpp │ ├── shrink.h │ ├── shrink_tiling_data.h │ └── shrink_tiling_key.h └── tests/ └── .gitkeep ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!3898 | 4 个月前 |