colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix the bug of pooling layer output shape mismatch

This commit is contained in:
zhengzhouheng 2021-02-24 10:01:12 +08:00 committed by hawk081
parent 9d44b4477b
commit c8b0180e7a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/tim/vx/ops/pool2d.h
View File

@ -38,7 +38,7 @@ class Pool2d : public Operation {
Pool2d(Graph* graph, PoolType type, PadType padding, Pool2d(Graph* graph, PoolType type, PadType padding,
const std::array<uint32_t, 2>& ksize, const std::array<uint32_t, 2>& ksize,
const std::array<uint32_t, 2>& stride, const std::array<uint32_t, 2>& stride,
RoundType round_type = RoundType::CEILING); RoundType round_type = RoundType::FLOOR);
protected: protected:
const PoolType type_; const PoolType type_;