文件最后提交记录最后更新时间
init 8 个月前
grouped_bias_add_grad算子原型增加输出约束描述 Co-authored-by: liaohuming<liaohuming@huawei.com> # message auto-generated for no-merge-commit merge: !2025 merge master into master grouped_bias_add_grad算子原型增加输出约束描述 Created-by: liaohuming Commit-by: liaohuming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1150 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 更新了grouped_bias_add_grad_proto.h文件 增加了输出grad_bias与输入grad_y的关系描述: grad_bias: A Tensor. Data type must match that of 'grad_y' (BFloat16, Float16 or Float32). Shape constraints are determined based on the dimensionality of 'grad_y': - If 'grad_y' has 3 dimensions ('group_idx' is not provided), then: 'grad_bias.shape = [grad_y.shape[0], grad_y.shape[2]]' - If 'grad_y' has 2 dimensions ('group_idx' is provided), then: 'grad_bias.shape = [group_idx.shape[0], grad_y.shape[1]]' - Supports non-continuous tensors, and data format supports "ND". ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!20252 个月前