| 15 | |
| 16 | using namespace MNN::Express; |
| 17 | static MNN::PoolPadType _convertPoollingPadMode(PaddingMode mode) { |
| 18 | switch (mode) { |
| 19 | case CAFFE: |
| 20 | return MNN::PoolPadType_CAFFE; |
| 21 | case VALID: |
| 22 | return MNN::PoolPadType_VALID; |
| 23 | case SAME: |
| 24 | return MNN::PoolPadType_SAME; |
| 25 | default: |
| 26 | break; |
| 27 | } |
| 28 | return MNN::PoolPadType_CAFFE; |
| 29 | } |
| 30 | static VARP _PoolGrad(VARP originInput, VARP originOutput, VARP inputGrad, INTS kernel, INTS stride, PoolingMode type, |
| 31 | PaddingMode pad = VALID, INTS pads= {0, 0}) { |
| 32 | std::unique_ptr<MNN::OpT> pool(new MNN::OpT); |