497bf5c9创建于 1 天前历史提交

文件	最后提交记录	最后更新时间
examples	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	17 天前
op_graph	补齐A5算子op_graph下面的cmakelist、同步修改apply_adam_w算子geir通路动态用例失败问题 Co-authored-by: t00850857<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !714 merge master into master 补齐A5算子op_graph下面的cmakelist、同步修改apply_adam_w算子geir通路动态用例失败问题 Created-by: tan_xin Commit-by: t00850857 Merged-by: cann-robot Description: ## 描述 1、补齐A5elu、gelu、gather_v2等算子op_graph下面的cmakelist 2、同步修改apply_adam_w算子geir通路动态用例失败问题 ## 关联的Issue [[DTS2025122942281](https://dts-szv.clouddragon.huawei.com/DTSPortal/ticket/DTS2025122942281)] ApplyAdamW geir通路功能有用例静态成功动态失败 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!714	6 个月前
op_host	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	17 天前
op_kernel	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	17 天前
tests	add_cases Co-authored-by: zhaogrug<zhaohai16@h-partners.com> # message auto-generated for no-merge-commit merge: !7583 merge add_cases into master add_cases Created-by: zhaogrug Commit-by: zhaogrug Merged-by: cann-robot Description: ## 描述新增仓内新版ttk用例 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue [#4223](https://gitcode.com/cann/ops-nn/issues/4223) <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试冒烟测试，本地验证 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：用例添加 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7583	1 天前
CMakeLists.txt	Arch编码更新 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !1116 merge ut_test into master Arch编码更新 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 Arch编码更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1116	5 个月前
README.md	nn仓doc tools扫描内容修改 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !3414 merge master into master nn仓doc tools扫描内容修改 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述修改markdown语法和链接，1000个左右。 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新更新了README.md、gitcode、cmake、common、control、conv、docs、examples、experimental、hash、index、loss、matmul、optim、pooling等文件。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!3414	3 个月前

AdamApplyOneWithDecay

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

算子功能：对模型中的一个参数（如权重），完成Adam优化算法的单步计算和更新。
计算公式：

$output0=input02×mul3_x+input1×mul2_xoutput_0 = input_0^2 \times mul_3\_x + input_1 \times mul_2\_x$

$output1=input2×mul0_x+input0×mul1_xoutput_1 = input_2 \times mul_0\_x + input_0 \times mul_1\_x$

$output2=input3−((output1(output0)+add2_y)+input3×mul4_x)×input4output_2 = input_3 - ((\frac{output_1}{\sqrt(output_0) + add_2\_y}) + input_3 \times mul_4\_x) \times input_4$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input0	输入	待进行adam_apply_one_with_decay计算的入参，公式中的input0。	BFLOAT16、FLOAT16、FLOAT	ND
input1	输入	待进行adam_apply_one_with_decay计算的入参，公式中的input1。	BFLOAT16、FLOAT16、FLOAT	ND
input2	输入	待进行adam_apply_one_with_decay计算的入参，公式中的input2。	BFLOAT16、FLOAT16、FLOAT	ND
input3	输入	待进行adam_apply_one_with_decay计算的入参，公式中的input3。	BFLOAT16、FLOAT16、FLOAT	ND
input4	输入	待进行adam_apply_one_with_decay计算的入参，公式中的input4。	BFLOAT16、FLOAT16、FLOAT	ND
mul0_x	输入	待进行adam_apply_one_with_decay计算的入参，公式中的mul0_x。	BFLOAT16、FLOAT16、FLOAT	ND
mul1_x	输入	待进行adam_apply_one_with_decay计算的入参，公式中的mul1_x。	BFLOAT16、FLOAT16、FLOAT	ND
mul2_x	输入	待进行adam_apply_one_with_decay计算的入参，公式中的mul2_x。	BFLOAT16、FLOAT16、FLOAT	ND
mul3_x	输入	待进行adam_apply_one_with_decay计算的入参，公式中的mul3_x。	BFLOAT16、FLOAT16、FLOAT	ND
mul4_x	输入	待进行adam_apply_one_with_decay计算的入参，公式中的mul4_x。	BFLOAT16、FLOAT16、FLOAT	ND
add2_y	输入	待进行adam_apply_one_with_decay计算的入参，公式中的add2_y。	BFLOAT16、FLOAT16、FLOAT	ND
output0	输出	待进行adam_apply_one_with_decay计算的出参，公式中的output0。	BFLOAT16、FLOAT16、FLOAT	ND
output1	输出	待进行adam_apply_one_with_decay计算的出参，公式中的output1。	BFLOAT16、FLOAT16、FLOAT	ND
output2	输出	待进行adam_apply_one_with_decay计算的出参，公式中的output2。	BFLOAT16、FLOAT16、FLOAT	ND

约束说明

无

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_adam_apply_one_with_decay	通过算子IR构图方式调用AdamApplyOne算子。