diff --git a/backends/test/suite/operators/test_avgpool1d.py b/backends/test/suite/operators/test_avgpool1d.py index 0b2d001de01..3aecfb55b0f 100644 --- a/backends/test/suite/operators/test_avgpool1d.py +++ b/backends/test/suite/operators/test_avgpool1d.py @@ -1,5 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. +# Copyright 2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -84,7 +85,7 @@ def test_avgpool1d_padding(self, flow: TestFlow) -> None: flow, ) self._test_op( - Model(padding=2), + Model(kernel_size=5, padding=2), (torch.randn(1, 8, 100),), flow, ) diff --git a/backends/test/suite/operators/test_avgpool2d.py b/backends/test/suite/operators/test_avgpool2d.py index 97bcb00372a..537146cc395 100644 --- a/backends/test/suite/operators/test_avgpool2d.py +++ b/backends/test/suite/operators/test_avgpool2d.py @@ -1,5 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. +# Copyright 2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -92,7 +93,7 @@ def test_avgpool2d_padding(self, flow: TestFlow) -> None: flow, ) self._test_op( - Model(padding=(1, 2)), + Model(kernel_size=5, padding=(1, 2)), (torch.randn(1, 8, 20, 20),), flow, ) diff --git a/backends/test/suite/operators/test_maxpool1d.py b/backends/test/suite/operators/test_maxpool1d.py index e6de4dee2b7..fdd6081e8af 100644 --- a/backends/test/suite/operators/test_maxpool1d.py +++ b/backends/test/suite/operators/test_maxpool1d.py @@ -1,5 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. +# Copyright 2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -86,7 +87,7 @@ def test_maxpool1d_padding(self, flow: TestFlow) -> None: flow, ) self._test_op( - Model(padding=2), + Model(kernel_size=5, padding=2), (torch.randn(1, 8, 100),), flow, ) diff --git a/backends/test/suite/operators/test_maxpool2d.py b/backends/test/suite/operators/test_maxpool2d.py index f8112d3b7da..9389a303139 100644 --- a/backends/test/suite/operators/test_maxpool2d.py +++ b/backends/test/suite/operators/test_maxpool2d.py @@ -1,5 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. +# Copyright 2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -91,7 +92,7 @@ def test_maxpool2d_padding(self, flow: TestFlow) -> None: flow, ) self._test_op( - Model(padding=(1, 2)), + Model(kernel_size=5, padding=(1, 2)), (torch.randn(1, 8, 20, 20),), flow, )