From 2bd4ca8377b59e3eeacb9c07ecd7d01b99fc9fb1 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Mon, 16 Mar 2020 05:58:17 +0900 Subject: [PATCH] add more tests --- tests/python/frontend/pytorch/test_forward.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 778177ebc6d1d..588ef2524502f 100644 --- 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():