Add limit tests for softmax

Author: irtrukhina

backends/nxp/backend/ir/converter/node_converters/ops_converters/softmax_converter.py

@@ -63,7 +63,7 @@ def _is_supported_on_target(
             """New flow: Hardware constraints for the new flow:
             1. Input and Output must be INT8/UINT8
             2. Channels <= 2040
-            3. Total spatial size (H*W) <= 4096
+            3. Total spatial size <= 4096
             4. Total size (channels * spatial_size) <= 524288
             """
             # Constraint 1: Input and Output must be INT8/UINT8.

backends/nxp/tests/ir/converter/node_converter/test_softmax_converter.py

@@ -248,7 +248,8 @@ def test__basic_nsys_inference(self, mocker, input_shape, dim):
             pytest.param((4096, 8), -1, id="2D_spatial_size_limit"),
             pytest.param((2040,), -1, id="1D_channels_limit"),
             pytest.param((4096, 128), -1, id="2D_total_size_limit"),
-            pytest.param((1, 64, 64, 8), -1, id="4D_spatial_size_limit"),
+            pytest.param((1, 64, 64, 8), -1, id="4D_spatial_size_limit_1x64x64"),
+            pytest.param((2, 32, 64, 8), -1, id="4D_spatial_size_limit_2x32x64"),
         ],
     )
     def test__limits(self, input_shape, dim, mocker):
@@ -266,7 +267,8 @@ def test__limits(self, input_shape, dim, mocker):
             pytest.param((4097, 8), -1, id="2D_spatial_size_exceeded"),
             pytest.param((2048,), -1, id="1D_channels_exceeded"),
             pytest.param((4096, 129), -1, id="2D_total_size_exceeded"),
-            pytest.param((1, 64, 65, 8), -1, id="4D_spatial_size_exceeded"),
+            pytest.param((1, 64, 65, 8), -1, id="4D_spatial_size_exceeded_1x64x65"),
+            pytest.param((2, 32, 65, 8), -1, id="4D_spatial_size_exceeded_2x32x65"),
         ],
     )
     def test__limits_exceeded(self, input_shape, dim):