Skip to content

Commit

Permalink
update model to use count_include_pad
Browse files Browse the repository at this point in the history
Signed-off-by: titaiwangms <[email protected]>
  • Loading branch information
titaiwangms committed Jan 23, 2025
1 parent 1dea833 commit 81e553a
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions onnx/backend/test/case/node/averagepool.py
Original file line number Diff line number Diff line change
Expand Up @@ -528,6 +528,7 @@ def export_averagepool_2d_ceil_last_window_starts_on_pad() -> None:
strides=[3, 3],
pads=[1, 1, 1, 1],
ceil_mode=True,
count_include_pad=1,
)
x = np.array(
[
Expand Down
Binary file not shown.

0 comments on commit 81e553a

Please sign in to comment.