已关闭
[Requirement|需求建议]: MAX_POOL3D_GRAD 算子新增NDHWC格式数据处理 #2335
dong-yanrong创建于  4月25日关闭于  4月25日
dong-yanrong成员
4月25日 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

MAX_POOL3D_GRAD 算子新增NDHWC格式数据处理,添加kernel部分simt的处理

Origin(信息来源)

NA

Benefit / Necessity (价值/作用)

最大池化通过取局部区域内的最大值来减少特征图的空间维度(即宽度和高度),而不改变通道数。具体来说,它类似于一个滑动窗口在输入特征图上滑动,对于每个局部区域,最大池化选择该区域内的最大数值作为输出结果。例如,对于一个2x2的池化核,如果输入是[[1,2], [3,4]],那么最大池化的输出就是4,因为4是这个区域内的最大值。

‌参数‌:最大池化操作通常包括两个主要参数:池化窗口大小(池化核大小)和步幅(stride)。

Design(设计方案)

1)在主调度逻辑中新增分支,以调用新的NDHWC内核。

likedislike
Ddong-yanrong成员
4月25日 添加了label:requirement
dong-yanrong成员
4月25日 评论:

/assign

likedislike
CANN-robotCANN-robot成员
4月25日 将 dong-yanrong 设为负责人
CANN-robotCANN-robot成员
4月25日 关闭了 issue
CANN-robotCANN-robot成员
4月25日 添加了label:resolved