onnx2versal
Loading...
Searching...
No Matches
Pool2D

Pool2D function on BCHW, yielding BCH'W', where H'=H/factor, W'=W/factor Scalar kernels allow W'<W/factor. More...

Collaboration diagram for Pool2D:

Modules

 Pool2DKernels
 

Classes

class  PoolGraph< POOL, TT, INP_H, INP_W, INP_W_PAD, OUT_H, OUT_W, B, C, KH, KW, STEP_H, STEP_W, PAD, H0, H1, W0, W1 >
 Single instance graph. More...
 

Detailed Description

Template Parameters
TTinput and output type
POOLPool Kernel
INP_Hinput height, used to calculate pool factor
INP_Winput width, used to calculate pool factor
INP_W_PADinput width, padded to vector boundary
OUT_Houtput height, used to calculate pool factor
OUT_Woutput width, used to calculate pool factor
Bbatch size
Cinput channels
KHkernel height
KWkernel width
STEP_Hstride in the height dimension
STEP_Wstride in the width dimension