add more tests

apache · Mar 15, 2020 · 2bd4ca8 · 2bd4ca8
1 parent c52962d
commit 2bd4ca8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py
@@ -707,6 +707,8 @@ def test_adaptive_pool3d():
     verify_model(torch.nn.AdaptiveMaxPool3d((2, 2, 2)).eval(), inp)
     verify_model(torch.nn.AdaptiveAvgPool3d((1, 1, 1)).eval(), inp)
     verify_model(torch.nn.AdaptiveAvgPool3d((2, 2, 2)).eval(), inp)
+    verify_model(torch.nn.AdaptiveAvgPool3d((4, 8, 8)).eval(), inp)
+    verify_model(torch.nn.AdaptiveMaxPool3d((7, 8, 9)).eval(), inp)
 
 
 def test_conv3d():