Pool2D function on BCHW, yielding BCH'W', where H'=H/factor, W'=W/factor Scalar kernels allow W'<W/factor.
More...
|
| 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...
|
| |
- Template Parameters
-
| TT | input and output type |
| POOL | Pool Kernel |
| INP_H | input height, used to calculate pool factor |
| INP_W | input width, used to calculate pool factor |
| INP_W_PAD | input width, padded to vector boundary |
| OUT_H | output height, used to calculate pool factor |
| OUT_W | output width, used to calculate pool factor |
| B | batch size |
| C | input channels |
| KH | kernel height |
| KW | kernel width |
| STEP_H | stride in the height dimension |
| STEP_W | stride in the width dimension |