Star0

Star0

cann-robot产品名变更

d679322b创建于 26 天前历史提交

文件	最后提交记录	最后更新时间
docs	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	26 天前
examples	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
op_api	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
op_host	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
op_kernel	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
tests	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
CMakeLists.txt	【社区任务】MinDim/MaxDim算子Ascend C实现 Co-authored-by: DaiVernon<205618093@qq.com> # message auto-generated for no-merge-commit merge: !4320 merge submission0725 into master 【社区任务】MinDim/MaxDim算子Ascend C实现 Created-by: 2302_79379917 Commit-by: DaiVernon Merged-by: cann-robot Description: ## 描述新增 MinDim/MaxDim 算子的 Ascend C 实现，支持 FLOAT16、FLOAT、BFLOAT16、INT16，以及正负规约轴、keepdim 和非连续 Tensor。 ## 关联的Issue Closes #2422 ## 测试已通过社区任务官方精度与性能验收。 ## 文档更新新增 MinDim/MaxDim README 和接口文档。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4320	2 个月前
README.md	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	26 天前

ArgMinWithValue

产品支持情况

产品	是否支持
Atlas A3系列产品	√
Atlas A2系列产品	√

功能说明

算子功能：在指定维度 dimension 上，返回输入张量 x 的最小值 values 及其 首次出现的下标 indice。等价于内置 aclnnMinDim（底层 IR 为 ArgMinWithValue）。
相等最小值取首个下标；支持 NaN 传播语义（与 torch.min(dim) 一致）。

参数说明

参数	输入/输出/属性	数据类型	数据格式	说明
x	输入	float16 / float / bfloat16 / int16	ND	1–8 维，支持非连续
dimension	属性	int	-	归约轴，[-x.dim(), x.dim())
keep_dims	属性	bool	-	是否保留归约轴，默认 false
indice	输出	int32	ND	最小值下标，与 values 同 shape
values	输出	同 x	ND	最小值，keep_dims=false 时降一维

约束说明

dimension 取值需在合法范围内。
规约轴长度必须大于 0；空规约轴属于未定义输入。

调用说明

调用方式	样例	说明
aclnn 接口	examples/test_aclnn_arg_min_with_value.cpp	5 组自检用例，覆盖 4 dtype、负轴、keepdim、ties、非连续输入