Add adaptive pooling tests for 3D models

openvinotoolkit · Dec 25, 2023 · 108d592 · 108d592
1 parent da3f73b
commit 108d592
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/frontends/paddle/tests/test_models/gen_scripts/generate_pool3d.py b/src/frontends/paddle/tests/test_models/gen_scripts/generate_pool3d.py
@@ -312,6 +312,12 @@ def main():
             "return_mask": False,
         }
         adaptive_pool3d(pooling_type + "AdaptivePool3D_test3", data_NCDHW, paddle_attrs)
+        paddle_attrs = {
+            "pool_size": 1,  # global pooling case
+            "pool_type": pooling_type,
+            "return_mask": True,
+        }
+        adaptive_pool3d(pooling_type + "AdaptivePool3D_test4", data_NCDHW, paddle_attrs)
 
 
 if __name__ == "__main__":