From c8b0180e7ae12535a38d1587da7b125141f8ed6f Mon Sep 17 00:00:00 2001
From: zhengzhouheng <zhengzhouheng@outlook.com>
Date: Wed, 24 Feb 2021 10:01:12 +0800
Subject: [PATCH] fix the bug of pooling layer output shape mismatch

---
 include/tim/vx/ops/pool2d.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/tim/vx/ops/pool2d.h b/include/tim/vx/ops/pool2d.h
index f4d0bc7..88a8a64 100644
--- a/include/tim/vx/ops/pool2d.h
+++ b/include/tim/vx/ops/pool2d.h
@@ -38,7 +38,7 @@ class Pool2d : public Operation {
   Pool2d(Graph* graph, PoolType type, PadType padding,
          const std::array<uint32_t, 2>& ksize,
          const std::array<uint32_t, 2>& stride,
-         RoundType round_type = RoundType::CEILING);
+         RoundType round_type = RoundType::FLOOR);
 
  protected:
   const PoolType type_;