add a new test

Author: titaiwangms

docs/Operators.md

@@ -1769,6 +1769,44 @@ expect(node, inputs=[x], outputs=[y], name="test_averagepool_2d_ceil")
 </details>
 
 
+<details>
+<summary>averagepool_2d_ceil_last_window_starts_on_pad</summary>
+
+```python
+"""input_shape: [1, 3, 2, 2]
+output_shape: [1, 3, 1, 1]
+"""
+node = onnx.helper.make_node(
+    "AveragePool",
+    inputs=["x"],
+    outputs=["y"],
+    kernel_shape=[3, 3],
+    strides=[3, 3],
+    pads=[1, 1, 1, 1],
+    ceil_mode=True,
+)
+x = np.array(
+    [
+        [
+            [[0.8580, 0.0786], [0.2692, 0.1537]],
+            [[0.8816, 0.4353], [0.5772, 0.6623]],
+            [[0.9067, 0.9483], [0.5970, 0.7630]],
+        ]
+    ]
+).astype(np.float32)
+y = np.array([[[[0.1511]], [[0.2841]], [[0.3572]]]]).astype(np.float32)
+
+expect(
+    node,
+    inputs=[x],
+    outputs=[y],
+    name="test_averagepool_2d_ceil_last_window_starts_on_pad",
+)
+```
+
+</details>
+
+
 <details>
 <summary>averagepool_2d_default</summary>
 

docs/TestCoverage.md

@@ -1105,7 +1105,7 @@ expect(node, inputs=[x], outputs=[y], name="test_atanh")
 
 
 ### AveragePool
-There are 16 test cases, listed as following:
+There are 17 test cases, listed as following:
 <details>
 <summary>averagepool_1d_default</summary>
 
@@ -1166,6 +1166,42 @@ y = np.array([[[[6, 7.5], [12, 13.5]]]]).astype(np.float32)
 expect(node, inputs=[x], outputs=[y], name="test_averagepool_2d_ceil")
 ```
 
+</details>
+<details>
+<summary>averagepool_2d_ceil_last_window_starts_on_pad</summary>
+
+```python
+"""input_shape: [1, 3, 2, 2]
+output_shape: [1, 3, 1, 1]
+"""
+node = onnx.helper.make_node(
+    "AveragePool",
+    inputs=["x"],
+    outputs=["y"],
+    kernel_shape=[3, 3],
+    strides=[3, 3],
+    pads=[1, 1, 1, 1],
+    ceil_mode=True,
+)
+x = np.array(
+    [
+        [
+            [[0.8580, 0.0786], [0.2692, 0.1537]],
+            [[0.8816, 0.4353], [0.5772, 0.6623]],
+            [[0.9067, 0.9483], [0.5970, 0.7630]],
+        ]
+    ]
+).astype(np.float32)
+y = np.array([[[[0.1511]], [[0.2841]], [[0.3572]]]]).astype(np.float32)
+
+expect(
+    node,
+    inputs=[x],
+    outputs=[y],
+    name="test_averagepool_2d_ceil_last_window_starts_on_pad",
+)
+```
+
 </details>
 <details>
 <summary>averagepool_2d_default</summary>

onnx/backend/test/case/node/averagepool.py

@@ -515,6 +515,38 @@ def export_averagepool_2d_ceil() -> None:
 
         expect(node, inputs=[x], outputs=[y], name="test_averagepool_2d_ceil")
 
+    @staticmethod
+    def export_averagepool_2d_ceil_last_window_starts_on_pad() -> None:
+        """input_shape: [1, 3, 2, 2]
+        output_shape: [1, 3, 1, 1]
+        """
+        node = onnx.helper.make_node(
+            "AveragePool",
+            inputs=["x"],
+            outputs=["y"],
+            kernel_shape=[3, 3],
+            strides=[3, 3],
+            pads=[1, 1, 1, 1],
+            ceil_mode=True,
+        )
+        x = np.array(
+            [
+                [
+                    [[0.8580, 0.0786], [0.2692, 0.1537]],
+                    [[0.8816, 0.4353], [0.5772, 0.6623]],
+                    [[0.9067, 0.9483], [0.5970, 0.7630]],
+                ]
+            ]
+        ).astype(np.float32)
+        y = np.array([[[[0.1511]], [[0.2841]], [[0.3572]]]]).astype(np.float32)
+
+        expect(
+            node,
+            inputs=[x],
+            outputs=[y],
+            name="test_averagepool_2d_ceil_last_window_starts_on_pad",
+        )
+
     @staticmethod
     def export_averagepool_2d_dilations() -> None:
         """input_shape: [1, 1, 4, 4]

onnx/backend/test/data/node/test_averagepool_2d_ceil_last_window_starts_on_pad/model.onnx

@@ -0,0 +1 @@
+

B
xJ0�[?	��=�ԉ>�c>��a?���>a�?~�)?~h?��r?��?�SC?

onnx/backend/test/data/node/test_averagepool_2d_ceil_last_window_starts_on_pad/test_data_set_0/input_0.pb

@@ -0,0 +1 @@
+



B
yJ��>�u�>��>