文件	最后提交记录	最后更新时间
docs	docs目录基本概念md中文名改为英文名 Co-authored-by: chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !7857 merge master into master docs目录基本概念md中文名改为英文名 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao;chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> docs目录基本概念md文件名改为英文：避免link中的中文字符引发的跳转异常，例如两段式接口.md变成%E4%B8%A4%E6%AE%B5%E5%BC%8F%E6%8E%A5%E5%8F%A3.md，不易于维护，可能导致其他平台跳转有问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#4260](https://gitcode.com/cann/ops-nn/issues/4260) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> docs/zh/context所有md和对应的link ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7857	26 天前
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	1 个月前
framework	TF plugin 迁移 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !7632 merge nn4 into master TF plugin 迁移 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 TF plugin 迁移 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4196 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7632	26 天前
op_api	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	1 个月前
op_graph	feat: 为缺失算子补充 op_graph 构建配置支持 Co-authored-by: cao-ren-lei<caorenlei@huawei.com> # message auto-generated for no-merge-commit merge: !4445 merge op_proto_new into master feat: 为缺失算子补充 op_graph 构建配置支持 Created-by: caorenlei Commit-by: cao-ren-lei Merged-by: cann-robot Description: ## 描述为缺失的算子补充 CMakeLists 配置文件，支持 op_graph 插件构建。 ### 改动原因部分算子的 op_graph 目录缺少 CMakeLists.txt 配置文件，导致无法正确编译 op_graph 插件。同时需要支持将 proto.h 文件合并到统一的 ops_proto_nn.h 文件中。 ### 改动方法 1. 新增构建配置文件：为 11 个算子的 op_graph 目录新增标准 CMakeLists.txt 文件，调用 `add_graph_plugin_sources()` 函数 2. 重命名 proto 头文件：将 `cross_entropy_loss.h` 和 `cross_entropy_loss_grad.h` 重命名为 `_proto.h`，支持合并到统一头文件 3. 修正文件格式：修复部分 CMakeLists.txt 文件末尾缺少换行符的问题 ### 涉及算子 - activation: softmax_cross_entropy_with_logits - loss: binary_cross_entropy, binary_cross_entropy_grad, cross_entropy_loss, cross_entropy_loss_grad, ctc_loss_v2, ctc_loss_v2_grad, l1_loss_grad, l2_loss, lp_loss, mse_loss, mse_loss_grad, smooth_l1_loss_grad_v2 - optim*: apply_adam_d ## 测试已验证 CMakeLists 配置文件的语法正确性，op_graph 插件能够正常编译。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：构建系统配置补充 ## 关联的Issue - #2550 See merge request: cann/ops-nn!4445	3 个月前
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	1 个月前
op_kernel	refactor: rename APIs in ops-nn batch4 (10 ops) Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !8569 merge rename-api-batch4 into master refactor: rename APIs in ops-nn batch4 (10 ops) Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述对 ops-nn 仓 10 个算子的 arch35 kernel 进行 AscendC MicroAPI 重命名整改，涉及以下 API 映射： - `__local_mem__` → `__ubuf__`（地址空间限定符） - `DataCopy`(load 方向) → `LoadAlign` - `DataCopy`(store 方向) → `StoreAlign` - `ReduceSum` → `Reduce<ReduceType::SUM>` - `DataCopyUnAlign` → `StoreUnAlign` - `DataCopyUnAlignPost` → `StoreUnAlignPost` - `UnalignReg` → `UnalignRegForStore` / `UnalignRegForLoad` 涉及算子（10 个）： - activation/softmax_cross_entropy_with_logits - activation/softmax_grad - activation/softmax_v2 - loss/nll_loss - loss/sparse_softmax_cross_entropy_with_logits - norm/rms_norm - norm/rms_norm_dynamic_mx_quant - norm/rms_norm_grad - norm/rms_norm_grad_quant - norm/rms_norm_quant_v2 共修改 38 个 arch35 kernel 头文件，纯 API 重命名，无逻辑变更。 ## 关联的Issue ## 测试纯 API 重命名整改，编译验证通过；无新增测试用例。 ## 文档更新无需更新文档。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：API 重命名重构 ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!8569	5 天前
tests	add cases Co-authored-by: zhaogrug<zhaohai16@h-partners.com> # message auto-generated for no-merge-commit merge: !8026 merge the_rest_of_cases into master add cases Created-by: zhaogrug Commit-by: zhaogrug Merged-by: cann-robot Description: ## 描述新增仓内新版ttk用例 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue [#4386](https://gitcode.com/cann/ops-nn/issues/4386) <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试冒烟测试，本地验证 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：用例添加 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8026	19 天前
CMakeLists.txt	feat: 为缺失算子补充 op_graph 构建配置支持 Co-authored-by: cao-ren-lei<caorenlei@huawei.com> # message auto-generated for no-merge-commit merge: !4445 merge op_proto_new into master feat: 为缺失算子补充 op_graph 构建配置支持 Created-by: caorenlei Commit-by: cao-ren-lei Merged-by: cann-robot Description: ## 描述为缺失的算子补充 CMakeLists 配置文件，支持 op_graph 插件构建。 ### 改动原因部分算子的 op_graph 目录缺少 CMakeLists.txt 配置文件，导致无法正确编译 op_graph 插件。同时需要支持将 proto.h 文件合并到统一的 ops_proto_nn.h 文件中。 ### 改动方法 1. 新增构建配置文件：为 11 个算子的 op_graph 目录新增标准 CMakeLists.txt 文件，调用 `add_graph_plugin_sources()` 函数 2. 重命名 proto 头文件：将 `cross_entropy_loss.h` 和 `cross_entropy_loss_grad.h` 重命名为 `_proto.h`，支持合并到统一头文件 3. 修正文件格式：修复部分 CMakeLists.txt 文件末尾缺少换行符的问题 ### 涉及算子 - activation: softmax_cross_entropy_with_logits - loss: binary_cross_entropy, binary_cross_entropy_grad, cross_entropy_loss, cross_entropy_loss_grad, ctc_loss_v2, ctc_loss_v2_grad, l1_loss_grad, l2_loss, lp_loss, mse_loss, mse_loss_grad, smooth_l1_loss_grad_v2 - optim*: apply_adam_d ## 测试已验证 CMakeLists 配置文件的语法正确性，op_graph 插件能够正常编译。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：构建系统配置补充 ## 关联的Issue - #2550 See merge request: cann/ops-nn!4445	3 个月前
README.md	Label the chip Co-authored-by: zhouwenfang<zhouwenfang3@h-partners.com> # message auto-generated for no-merge-commit merge: !7823 merge master into master Label the chip Created-by: zhouwenfang Commit-by: zhouwenfang Merged-by: cann-robot Description: ## 描述 Label the chip ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/4245 ## 测试 NA ## 文档更新 activation目录所有的aclnn.md文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7823	26 天前

SoftmaxCrossEntropyWithLogits

产品支持情况

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

功能说明

接口功能：计算softmax和cross entropy的交叉熵损失，并给出对输入logits的反向梯度。
计算公式：
$loss=−∑iyi⋅log⁡(softmax(xi))\text{loss} = - \sum_i y_i \cdot \log(\text{softmax}(x_i))$

backprop=softmax(xi)−yi\text{backprop} = \text{softmax}(x_i) - y_i

其中， $x_i$ 对应输入的features， $y_i$ 对应输入的labels。

参数说明

参数名	输入/输出	描述	数据类型	数据格式
features	输入	表示模型的输出logits，公式中x_i。	FLOAT、FLOAT16、BFLOAT16	ND
labels	输入	表示输入的labels，公式中y_i。	FLOAT、FLOAT16、BFLOAT16	ND
loss	输出	表示softmax和cross entropy的交叉熵损失，公式中loss。	FLOAT、FLOAT16、BFLOAT16	ND
backprop	输出	表示输入logits的反向梯度，公式中backprop。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_softmax_cross_entropy_with_logits	通过aclnnSoftmaxCrossEntropyWithLogits接口方式调用SoftmaxCrossEntropyWithLogits算子。
图模式	-	通过算子IR构图方式调用SoftmaxCrossEntropyWithLogits算子。