Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
MAX_POOL3D_GRAD 算子新增NDHWC格式数据处理,添加kernel部分simt的处理
NA
最大池化通过取局部区域内的最大值来减少特征图的空间维度(即宽度和高度),而不改变通道数。具体来说,它类似于一个滑动窗口在输入特征图上滑动,对于每个局部区域,最大池化选择该区域内的最大数值作为输出结果。例如,对于一个2x2的池化核,如果输入是[[1,2], [3,4]],那么最大池化的输出就是4,因为4是这个区域内的最大值。
参数:最大池化操作通常包括两个主要参数:池化窗口大小(池化核大小)和步幅(stride)。
1)在主调度逻辑中新增分支,以调用新的NDHWC内核。
/assign
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内核。