diff --git a/tests/onnx/test_auto_scan_averagepool_10.py b/tests/onnx/test_auto_scan_averagepool_10.py index bfe35e07dbdccb1f17dc410a300b1f914974e914..1e569a2f5d84ff707e2dd0d5ee6141b5ab6a80e9 100644 --- a/tests/onnx/test_auto_scan_averagepool_10.py +++ b/tests/onnx/test_auto_scan_averagepool_10.py @@ -30,8 +30,9 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): input_shape = draw( st.lists( st.integers( - min_value=10, max_value=20), min_size=4, max_size=4)) - + min_value=20, max_value=30), min_size=4, max_size=4)) + # BS = 1 + input_shape[0] = 1 kernel_size = draw( st.lists( st.integers( @@ -40,7 +41,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): strides = draw( st.lists( st.integers( - min_value=1, max_value=5), min_size=2, max_size=2)) + min_value=1, max_value=2), min_size=2, max_size=2)) if draw(st.booleans()): auto_pad = "NOTSET" @@ -96,7 +97,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): return (config, attrs) def test(self): - self.run_and_statis(max_examples=30) + self.run_and_statis(max_examples=300) if __name__ == "__main__": diff --git a/tests/onnx/test_auto_scan_averagepool_7.py b/tests/onnx/test_auto_scan_averagepool_7.py index e0a68cf11f35340cb8f32e25a81b48ecad6e6bbe..666800bf47d0c1aab0975a681aae5f34bafa6859 100644 --- a/tests/onnx/test_auto_scan_averagepool_7.py +++ b/tests/onnx/test_auto_scan_averagepool_7.py @@ -40,7 +40,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): strides = draw( st.lists( st.integers( - min_value=1, max_value=5), min_size=2, max_size=2)) + min_value=1, max_value=2), min_size=2, max_size=2)) if draw(st.booleans()): auto_pad = "NOTSET"