描述

本 PR 为 Ascend 950 的 BNTrainingReduce GE 内核补充 NCDHW 5 维格式支持。算子保留 C 轴并归约其余维度；对 NCDHW 输入沿 N、D、H、W 轴计算每通道 sum 与 square_sum。

主要改动：

• 在 OpDef 中为 FLOAT16、BFLOAT16、FLOAT 输入补充 NCDHW 格式组合，输出固定为 FLOAT，并保持 GE 输出格式元数据与输入一致。
• 扩展 InferShape：支持 NCHW 2～4 维、NHWC 4 维和 NCDHW 5 维；NCHW/NCDHW 使用第 1 维作为 C 轴，NHWC 使用第 3 维；拒绝 NDC1HWC0 及格式/Rank 不匹配输入。
• 扩展 Ascend 950 Tiling 公共输入结构、格式枚举和校验逻辑，新增 NCDHW 路由，并兼容逻辑输出 [C] 与仅通道维非 1 的展开输出。
• 保持 ACLNN 接口仅支持 NCHW 的既有契约；将一维输出在 AICore 调用前扩展为 NCHW 四维视图，同时补充空 Tensor 与格式校验。
• 确定性模式下改用用户 workspace，避免将框架 workspace 直接传入算子实现。
• 更新 GEIR 示例、README 及相关 Host UT。

关联的Issue

• https://gitcode.com/cann/ops-nn/issues/5085

测试

• InferShape UT：覆盖 NCHW 2 维、NCDHW 5 维、NHWC 4 维，以及 ND、NDC1HWC0、NHWC/NCDHW 错误 Rank 拒绝场景。
• Tiling UT：覆盖 NCHW 2 维、NHWC、NCDHW、空通道/空归约轴、非法格式、错误 Rank 和输出 Shape 不匹配场景。
• ACLNN UT：覆盖 BF16 正向场景，并验证 NHWC 与空 NHWC 输入被拒绝，确保 ACLNN 契约无扩张。
• GEIR 示例：增加 NCDHW 格式解析与输出格式打印支持。
• GitCode CI 已通过：antipoison、codecheck、codespell、链接有效性、资源存在性、标签闭合、markdownlint、pre-commit 和 SCA 检查；当前 PR 标签为 ci-pipeline-passed。

文档更新

更新 norm/bn_training_reduce/README.md：补充 Ascend 950 GE 内核支持 NCDHW 5 维、各格式 Rank 范围、输出 Shape/格式约束，以及 ACLNN 与 GE 通路的格式差异。

类型标签

• Bug修复
• 新特性
• 性能优化
• 文档更新
• 其他，请描述：

AI/Agent生成声明

• AI辅助编写