Star0

Star0

cann-roboterrMsg对算子MaxPoolV3,MaxPoolV2,max_pool_grad_with_argmax做整改，当前修改tiling和infershape

4470e086创建于 6月8日历史提交

文件	最后提交记录	最后更新时间
examples	docs: add README and example for max_pool_grad_with_argmax operator Co-authored-by: liuchuangdev<liuchuang51@huawei.com> # message auto-generated for no-merge-commit merge: !4829 merge docs-max_pool_grad_with_argmax into master docs: add README and example for max_pool_grad_with_argmax operator Created-by: liuchuangdev Commit-by: liuchuangdev Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> max_pool_grad_with_argmax文档补齐，增加readme和example ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1602](https://gitcode.com/cann/ops-nn/issues/1602) [#2027](https://gitcode.com/cann/ops-nn/issues/2027) [#2028](https://gitcode.com/cann/ops-nn/issues/2028) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> docs/zh/op_list.md pooling/max_pool_grad_with_argmax/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4829	3 个月前
op_graph	[CANNBot]新增pooling融合规则并迁移至新Pass框架 Co-authored-by: Apricityh<wangxu359@huawei.com> # message auto-generated for no-merge-commit merge: !2811 merge FusionPass into master [CANNBot]新增pooling融合规则并迁移至新Pass框架 Created-by: Apricityh Commit-by: Apricityh Merged-by: cann-robot Description: ## 描述 [CANNBot]新增pooling融合规则并迁移至新Pass框架 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/2248 ## 测试回归图转换成功且通路算子精度pass ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：pooling融合规则并迁移至新Pass框架 ## 改动说明基于新Pass框架与ES接口重构并迁移以下融合规则： 1. MaxPoolFusionPass: MaxPool → MaxPoolV3 2. MaxPoolWithArgmaxV3FusionPass: V1/V2 → MaxPoolWithArgmaxV3 3. MaxPoolGradWithArgmaxV3FusionPass: V1/V2 → MaxPoolGradWithArgmaxV3 ## AI/Agent 生成声明代码来源： - [ ] 纯人工手写 - [x] AI 辅助编写（思路/片段/优化） - [ ] AI 完全生成人工审查：已完成测试验证：已通过合规检查：已完成 See merge request: cann/ops-nn!2811	4 个月前
op_host	errMsg对算子MaxPoolV3,MaxPoolV2,max_pool_grad_with_argmax做整改，当前修改tiling和infershape Co-authored-by: duxinlei<duxinlei@h-partners.com> # message auto-generated for no-merge-commit merge: !5564 merge errMsg into master errMsg对算子MaxPoolV3,MaxPoolV2,max_pool_grad_with_argmax做整改，当前修改tiling和infershape Created-by: duxinlei Commit-by: duxinlei Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5564	3 个月前
op_kernel	池化算子提取反向算子公共文件夹 Co-authored-by: liuchuangdev<liuchuang51@huawei.com> # message auto-generated for no-merge-commit merge: !2251 merge refactor into master 池化算子提取反向算子公共文件夹 Created-by: liuchuangdev Commit-by: liuchuangdev Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 池化算子结构优化：提取反向算子公共文件夹、优化算子目录结构 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1319](https://gitcode.com/cann/ops-nn/issues/1319) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基本功能回归、冒烟测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码结构重构 See merge request: cann/ops-nn!2251	6 个月前
tests	池化算子提取反向算子公共文件夹 Co-authored-by: liuchuangdev<liuchuang51@huawei.com> # message auto-generated for no-merge-commit merge: !2251 merge refactor into master 池化算子提取反向算子公共文件夹 Created-by: liuchuangdev Commit-by: liuchuangdev Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 池化算子结构优化：提取反向算子公共文件夹、优化算子目录结构 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1319](https://gitcode.com/cann/ops-nn/issues/1319) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基本功能回归、冒烟测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码结构重构 See merge request: cann/ops-nn!2251	6 个月前
CMakeLists.txt	池化算子提取反向算子公共文件夹 Co-authored-by: liuchuangdev<liuchuang51@huawei.com> # message auto-generated for no-merge-commit merge: !2251 merge refactor into master 池化算子提取反向算子公共文件夹 Created-by: liuchuangdev Commit-by: liuchuangdev Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 池化算子结构优化：提取反向算子公共文件夹、优化算子目录结构 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1319](https://gitcode.com/cann/ops-nn/issues/1319) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基本功能回归、冒烟测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码结构重构 See merge request: cann/ops-nn!2251	6 个月前
README.md	docs: add README and example for max_pool_grad_with_argmax operator Co-authored-by: liuchuangdev<liuchuang51@huawei.com> # message auto-generated for no-merge-commit merge: !4829 merge docs-max_pool_grad_with_argmax into master docs: add README and example for max_pool_grad_with_argmax operator Created-by: liuchuangdev Commit-by: liuchuangdev Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> max_pool_grad_with_argmax文档补齐，增加readme和example ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1602](https://gitcode.com/cann/ops-nn/issues/1602) [#2027](https://gitcode.com/cann/ops-nn/issues/2027) [#2028](https://gitcode.com/cann/ops-nn/issues/2028) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> docs/zh/op_list.md pooling/max_pool_grad_with_argmax/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4829	3 个月前

MaxPoolGradWithArgmax

产品支持情况

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

功能说明

算子功能：正向最大池化MaxPoolWithArgmax的反向梯度计算。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入x，形状为[NCHW]或[NHWC]	FLOAT16、BFLOAT16、FLOAT	NCHW、NHWC
grad	输入	梯度Tensor，形状为[NCHW]或[NHWC]	FLOAT16、BFLOAT16、FLOAT	NCHW、NHWC
argmax	输入	正向最大池化输出的索引，形状与grad相同	INT32、INT64	NCHW、NHWC
ksize	属性（必选）	池化窗口大小，长度为4的列表。NCHW格式时ksize[0]=1且ksize[1]=1；NHWC格式时ksize[0]=1且ksize[3]=1	ListInt	ND
strides	属性（必选）	窗口移动步长，长度为4的列表。NCHW格式时strides[0]=1且strides[1]=1；NHWC格式时strides[0]=1且strides[3]=1	ListInt	ND
padding	属性（必选）	填充算法，取值为"SAME"或"VALID"	String	ND
include_batch_in_index	属性（可选）	是否在计算argmax索引时包含batch维度。当前仅支持false。默认值：false	Bool	ND
data_format	属性（可选）	数据布局格式，取值为"NHWC"或"NCHW"。默认值："NHWC"	String	ND
y	输出	输出梯度，形状与x相同	FLOAT16、BFLOAT16、FLOAT	NCHW、NHWC

约束说明

include_batch_in_index：当前仅支持false

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_max_pool_grad_with_argmax	通过IR MaxPoolGradWithArgmax构图方式调用MaxPoolGradWithArgmax算子。