From ba34fa77e8adc529036bcd9397485f19a616a2d2 Mon Sep 17 00:00:00 2001
From: Anastasia Kuporosova <anastasia.kuporosova@intel.com>
Date: Fri, 8 Dec 2023 14:57:11 +0100
Subject: [PATCH] [PyOV] Expose result op (#21482)

* [PyOV] Expose result op

* update docs with op.Result

* codestyle

* add test and fix flake8 errors

* add result test

* fix transformation tests

* update return type
---
 .../src/openvino/runtime/op/__init__.py       |  3 +-
 .../src/openvino/runtime/opset13/__init__.py  |  2 +-
 .../src/openvino/runtime/opset13/ops.py       | 16 ++++-
 .../python/src/pyopenvino/graph/model.cpp     | 18 +++---
 .../python/src/pyopenvino/graph/ops/if.cpp    |  4 +-
 .../src/pyopenvino/graph/ops/result.cpp       |  2 +
 .../python/tests/test_graph/test_basic.py     | 10 +--
 .../python/tests/test_graph/test_manager.py   | 19 +++---
 .../python/tests/test_graph/test_ops.py       | 62 ++++++++++---------
 .../tests/test_graph/test_ops_result.py       | 28 +++++++++
 .../python/tests/test_graph/test_pad.py       | 27 +++-----
 11 files changed, 112 insertions(+), 79 deletions(-)
 create mode 100644 src/bindings/python/tests/test_graph/test_ops_result.py

diff --git a/src/bindings/python/src/openvino/runtime/op/__init__.py b/src/bindings/python/src/openvino/runtime/op/__init__.py
index 069976998f4..32ec2e8b909 100644
--- a/src/bindings/python/src/openvino/runtime/op/__init__.py
+++ b/src/bindings/python/src/openvino/runtime/op/__init__.py
@@ -8,11 +8,10 @@ Low level wrappers for the c++ api in ov::op.
 
 # flake8: noqa
 
-import numpy as np
-
 from openvino._pyopenvino.op import Constant
 from openvino._pyopenvino.op import assign
 from openvino._pyopenvino.op import Parameter
 from openvino._pyopenvino.op import if_op
 from openvino._pyopenvino.op import loop
 from openvino._pyopenvino.op import tensor_iterator
+from openvino._pyopenvino.op import Result
diff --git a/src/bindings/python/src/openvino/runtime/opset13/__init__.py b/src/bindings/python/src/openvino/runtime/opset13/__init__.py
index 9c544b0d7e7..1abac3e07a7 100644
--- a/src/bindings/python/src/openvino/runtime/opset13/__init__.py
+++ b/src/bindings/python/src/openvino/runtime/opset13/__init__.py
@@ -144,7 +144,7 @@ from openvino.runtime.opset1.ops import region_yolo
 from openvino.runtime.opset2.ops import reorg_yolo
 from openvino.runtime.opset1.ops import relu
 from openvino.runtime.opset1.ops import reshape
-from openvino.runtime.opset1.ops import result
+from openvino.runtime.opset13.ops import result
 from openvino.runtime.opset1.ops import reverse_sequence
 from openvino.runtime.opset3.ops import rnn_cell
 from openvino.runtime.opset5.ops import rnn_sequence
diff --git a/src/bindings/python/src/openvino/runtime/opset13/ops.py b/src/bindings/python/src/openvino/runtime/opset13/ops.py
index 27895535ead..6d4864c7aa1 100644
--- a/src/bindings/python/src/openvino/runtime/opset13/ops.py
+++ b/src/bindings/python/src/openvino/runtime/opset13/ops.py
@@ -11,8 +11,8 @@ import numpy as np
 
 log = logging.getLogger(__name__)
 
-from openvino.runtime import Node, Shape, Type
-from openvino.runtime.op import Constant
+from openvino.runtime import Node, Shape, Type, Output
+from openvino.runtime.op import Constant, Result
 from openvino.runtime.opset_utils import _get_node_factory
 from openvino.runtime.utils.decorators import binary_op, nameable_op, unary_op
 from openvino.runtime.utils.types import (
@@ -334,3 +334,15 @@ def constant(
                     _value, _shared_memory = _value.astype(_dtype), False
     # Create Constant itself:
     return Constant(_value, shared_memory=_shared_memory)
+
+
+@unary_op
+def result(data: Union[Node, Output, NumericData], name: Optional[str] = None) -> Node:
+    """Return a node which represents an output of a graph (Model).
+
+    :param data: The tensor containing the input data
+    :return: Result node
+    """
+    if isinstance(data, Node):
+        return Result(data.output(0))
+    return Result(data)
diff --git a/src/bindings/python/src/pyopenvino/graph/model.cpp b/src/bindings/python/src/pyopenvino/graph/model.cpp
index 6f5e1d2ff3d..374bbe8704e 100644
--- a/src/bindings/python/src/pyopenvino/graph/model.cpp
+++ b/src/bindings/python/src/pyopenvino/graph/model.cpp
@@ -111,7 +111,7 @@ void regclass_graph_Model(py::module m) {
                     Create user-defined Model which is a representation of a model.
 
                     :param results: List of results.
-                    :type results: List[openvino.runtime.Node]
+                    :type results: List[op.Result]
                     :param sinks: List of Nodes to be used as Sinks (e.g. Assign ops).
                     :type sinks: List[openvino.runtime.Node]
                     :param parameters: List of parameters.
@@ -221,7 +221,7 @@ void regclass_graph_Model(py::module m) {
             Create user-defined Model which is a representation of a model
 
             :param results: List of results.
-            :type results: List[openvino.runtime.Node]
+            :type results: List[op.Result]
             :param sinks: List of Nodes to be used as Sinks (e.g. Assign ops).
             :type sinks: List[openvino.runtime.Node]
             :param parameters: List of parameters.
@@ -274,7 +274,7 @@ void regclass_graph_Model(py::module m) {
             Create user-defined Model which is a representation of a model
 
             :param results: List of results.
-            :type results: List[openvino.runtime.Node]
+            :type results: List[op.Result]
             :param parameters: List of parameters.
             :type parameters: List[op.Parameter]
             :param variables: List of variables.
@@ -538,7 +538,7 @@ void regclass_graph_Model(py::module m) {
                     Return a list of model outputs.
 
                     :return: a list of model's result nodes.
-                    :rtype: List[openvino.runtime.Node]
+                    :rtype: List[op.Result]
                 )");
     model.def_property_readonly("results",
                                 &ov::Model::get_results,
@@ -546,7 +546,7 @@ void regclass_graph_Model(py::module m) {
                                     Return a list of model outputs.
 
                                     :return: a list of model's result nodes.
-                                    :rtype: List[openvino.runtime.Node]
+                                    :rtype: List[op.Result]
                                 )");
     model.def("get_result",
               &ov::Model::get_result,
@@ -554,7 +554,7 @@ void regclass_graph_Model(py::module m) {
                     Return single result.
 
                     :return: Node object representing result.
-                    :rtype: openvino.runtime.Node
+                    :rtype: op.Result
                 )");
     model.def_property_readonly("result",
                                 &ov::Model::get_result,
@@ -562,7 +562,7 @@ void regclass_graph_Model(py::module m) {
                                         Return single result.
 
                                         :return: Node object representing result.
-                                        :rtype: openvino.runtime.Node
+                                        :rtype: op.Result
                                     )");
     model.def("get_result_index",
               (int64_t(ov::Model::*)(const ov::Output<ov::Node>&) const) & ov::Model::get_result_index,
@@ -747,7 +747,7 @@ void regclass_graph_Model(py::module m) {
                 Delete Result node from the list of results. Method will not delete node from graph.
 
                 :param result: Result node to delete.
-                :type result: openvino.runtime.Node
+                :type result: op.Result
             )");
 
     model.def("remove_parameter",
@@ -827,7 +827,7 @@ void regclass_graph_Model(py::module m) {
                     Method doesn't validate graph, it should be done manually after all changes.
 
                     :param results: new Result nodes.
-                    :type results: List[openvino.runtime.Node]
+                    :type results: List[op.Result]
                 )");
 
     model.def(
diff --git a/src/bindings/python/src/pyopenvino/graph/ops/if.cpp b/src/bindings/python/src/pyopenvino/graph/ops/if.cpp
index ad83891ebaa..10453341e38 100644
--- a/src/bindings/python/src/pyopenvino/graph/ops/if.cpp
+++ b/src/bindings/python/src/pyopenvino/graph/ops/if.cpp
@@ -115,10 +115,10 @@ void regclass_graph_op_If(py::module m) {
             Sets new output from the operation associated with results of each sub-graphs.
 
             :param then_result: result from then_body.
-            :type then_result: openvino.runtime.Node
+            :type then_result: op.Result
 
             :param else_result: result from else_body.
-            :type else_result: openvino.runtime.Node
+            :type else_result: op.Result
 
             :return: output from operation.
             :rtype: openvino.runtime.Output
diff --git a/src/bindings/python/src/pyopenvino/graph/ops/result.cpp b/src/bindings/python/src/pyopenvino/graph/ops/result.cpp
index 022982a2799..c8576c8ecaf 100644
--- a/src/bindings/python/src/pyopenvino/graph/ops/result.cpp
+++ b/src/bindings/python/src/pyopenvino/graph/ops/result.cpp
@@ -20,6 +20,8 @@ void regclass_graph_op_Result(py::module m) {
 
     result.doc() = "openvino.runtime.op.Result wraps ov::op::v0::Result";
 
+    result.def(py::init<const ov::Output<ov::Node>&>());
+
     result.def("get_output_partial_shape", &ov::Node::get_output_partial_shape, py::arg("index"));
 
     result.def("get_output_element_type", &ov::Node::get_output_element_type, py::arg("index"));
diff --git a/src/bindings/python/tests/test_graph/test_basic.py b/src/bindings/python/tests/test_graph/test_basic.py
index 1c5c8b10bbe..45723c6db7b 100644
--- a/src/bindings/python/tests/test_graph/test_basic.py
+++ b/src/bindings/python/tests/test_graph/test_basic.py
@@ -25,6 +25,7 @@ from openvino.runtime import AxisVector, Coordinate, CoordinateDiff
 from openvino._pyopenvino import DescriptorTensor
 
 from openvino.runtime.utils.types import get_element_type
+from tests.utils.helpers import generate_model_with_memory
 
 
 def test_graph_api():
@@ -554,12 +555,7 @@ def test_multiple_outputs():
 
 
 def test_sink_model_ctor():
-    input_data = ops.parameter([2, 2], name="input_data", dtype=np.float32)
-    rv = ops.read_value(input_data, "var_id_667", np.float32, [2, 2])
-    add = ops.add(rv, input_data, name="MemoryAdd")
-    node = ops.assign(add, "var_id_667")
-    res = ops.result(add, "res")
-    model = Model(results=[res], sinks=[node], parameters=[input_data], name="TestModel")
+    model = generate_model_with_memory(input_shape=[2, 2], data_type=np.float32)
 
     ordered_ops = model.get_ordered_ops()
     op_types = [op.get_type_name() for op in ordered_ops]
@@ -570,7 +566,7 @@ def test_sink_model_ctor():
     assert len(model.get_ops()) == 5
     assert model.get_output_size() == 1
     assert model.get_output_op(0).get_type_name() == "Result"
-    assert model.get_output_element_type(0) == input_data.get_element_type()
+    assert model.get_output_element_type(0) == model.get_parameters()[0].get_element_type()
     assert list(model.get_output_shape(0)) == [2, 2]
     assert (model.get_parameters()[0].get_partial_shape()) == PartialShape([2, 2])
     assert len(model.get_parameters()) == 1
diff --git a/src/bindings/python/tests/test_graph/test_manager.py b/src/bindings/python/tests/test_graph/test_manager.py
index 3a1b7740ab8..95be2a1df2c 100644
--- a/src/bindings/python/tests/test_graph/test_manager.py
+++ b/src/bindings/python/tests/test_graph/test_manager.py
@@ -1,8 +1,7 @@
+# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-# flake8: noqa
-
 import os
 
 import numpy as np
@@ -15,6 +14,7 @@ from openvino.runtime.passes import Manager, Serialize, ConstantFolding, Version
 from tests.test_graph.util import count_ops_of_type
 from tests.utils.helpers import create_filename_for_test, compare_models
 
+
 def create_model():
     shape = [100, 100, 2]
     parameter_a = ops.parameter(shape, dtype=np.float32, name="A")
@@ -40,7 +40,8 @@ def test_constant_folding():
     assert count_ops_of_type(model, node_ceil) == 0
     assert count_ops_of_type(model, node_constant) == 1
 
-    new_const = model.get_results()[0].input(0).get_source_output().get_node()
+    result = model.get_results()[0]
+    new_const = result.input(0).get_source_output().get_node()
     values_out = new_const.get_vector()
     values_expected = [0.0, 1.0, 0.0, -2.0, 3.0, 3.0]
 
@@ -48,14 +49,14 @@ def test_constant_folding():
 
 
 # request - https://docs.pytest.org/en/7.1.x/reference/reference.html#request
-@pytest.fixture
+@pytest.fixture()
 def prepare_ir_paths(request, tmp_path):
     xml_path, bin_path = create_filename_for_test(request.node.name, tmp_path)
 
     yield xml_path, bin_path
-    
+
     # IR Files deletion should be done after `Model` is destructed.
-    # It may be achieved by splitting scopes (`Model` will be destructed 
+    # It may be achieved by splitting scopes (`Model` will be destructed
     # just after test scope finished), or by calling `del Model`
     os.remove(xml_path)
     os.remove(bin_path)
@@ -104,7 +105,7 @@ def test_serialize_separate_paths_args(prepare_ir_paths):
 
 def test_serialize_pass_mixed_args_kwargs(prepare_ir_paths):
     core = Core()
-    
+
     shape = [3, 2]
     parameter_a = ops.parameter(shape, dtype=np.float32, name="A")
     parameter_b = ops.parameter(shape, dtype=np.float32, name="B")
@@ -123,7 +124,7 @@ def test_serialize_pass_mixed_args_kwargs(prepare_ir_paths):
 
 def test_serialize_pass_mixed_args_kwargs_v2(prepare_ir_paths):
     core = Core()
-    
+
     xml_path, bin_path = prepare_ir_paths
     model = create_model()
     pass_manager = Manager()
@@ -175,7 +176,7 @@ def test_default_version(prepare_ir_paths):
     assert compare_models(model, res_model)
 
 
-def test_default_version_IR_V11_separate_paths(prepare_ir_paths):
+def test_default_version_ir_v11_separate_paths(prepare_ir_paths):
     core = Core()
 
     xml_path, bin_path = prepare_ir_paths
diff --git a/src/bindings/python/tests/test_graph/test_ops.py b/src/bindings/python/tests/test_graph/test_ops.py
index 6591cb8f577..433b798dec0 100644
--- a/src/bindings/python/tests/test_graph/test_ops.py
+++ b/src/bindings/python/tests/test_graph/test_ops.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
@@ -12,6 +13,7 @@ from openvino import Shape, Type
 from openvino.runtime import AxisSet
 from openvino.runtime.op import Constant, Parameter
 
+
 @pytest.mark.parametrize(("ov_op", "expected_ov_str", "expected_type"), [
     (lambda a, b: a + b, "Add", Type.f32),
     (ov.add, "Add", Type.f32),
@@ -34,9 +36,9 @@ from openvino.runtime.op import Constant, Parameter
 def test_binary_op(ov_op, expected_ov_str, expected_type):
     element_type = Type.f32
     shape = Shape([2, 2])
-    A = Parameter(element_type, shape)
-    B = Parameter(element_type, shape)
-    node = ov_op(A, B)
+    param1 = Parameter(element_type, shape)
+    param2 = Parameter(element_type, shape)
+    node = ov_op(param1, param2)
 
     assert node.get_type_name() == expected_ov_str
     assert node.get_output_size() == 1
@@ -48,10 +50,10 @@ def test_add_with_mul():
 
     element_type = Type.f32
     shape = Shape([4])
-    A = Parameter(element_type, shape)
-    B = Parameter(element_type, shape)
-    C = Parameter(element_type, shape)
-    node = ov.multiply(ov.add(A, B), C)
+    param1 = Parameter(element_type, shape)
+    param2 = Parameter(element_type, shape)
+    param3 = Parameter(element_type, shape)
+    node = ov.multiply(ov.add(param1, param2), param3)
 
     assert node.get_type_name() == "Multiply"
     assert node.get_output_size() == 1
@@ -85,8 +87,8 @@ def test_unary_op(ov_op, expected_ov_str):
 
     element_type = Type.f32
     shape = Shape([4])
-    A = Parameter(element_type, shape)
-    node = ov_op(A)
+    param1 = Parameter(element_type, shape)
+    node = ov_op(param1)
 
     assert node.get_type_name() == expected_ov_str
     assert node.get_output_size() == 1
@@ -97,8 +99,8 @@ def test_unary_op(ov_op, expected_ov_str):
 def test_reshape():
     element_type = Type.f32
     shape = Shape([2, 3])
-    A = Parameter(element_type, shape)
-    node = ov.reshape(A, Shape([3, 2]), special_zero=False)
+    param1 = Parameter(element_type, shape)
+    node = ov.reshape(param1, Shape([3, 2]), special_zero=False)
 
     assert node.get_type_name() == "Reshape"
     assert node.get_output_size() == 1
@@ -108,8 +110,8 @@ def test_reshape():
 
 def test_broadcast():
     element_type = Type.f32
-    A = Parameter(element_type, Shape([3]))
-    node = ov.broadcast(A, [3, 3])
+    param1 = Parameter(element_type, Shape([3]))
+    node = ov.broadcast(param1, [3, 3])
     assert node.get_type_name() == "Broadcast"
     assert node.get_output_size() == 1
     assert list(node.get_output_shape(0)) == [3, 3]
@@ -134,10 +136,10 @@ def test_constant(const, args, expectation):
 
 def test_concat():
     element_type = Type.f32
-    A = Parameter(element_type, Shape([1, 2]))
-    B = Parameter(element_type, Shape([1, 2]))
-    C = Parameter(element_type, Shape([1, 2]))
-    node = ov.concat([A, B, C], axis=0)
+    param1 = Parameter(element_type, Shape([1, 2]))
+    param2 = Parameter(element_type, Shape([1, 2]))
+    param3 = Parameter(element_type, Shape([1, 2]))
+    node = ov.concat([param1, param2, param3], axis=0)
     assert node.get_type_name() == "Concat"
     assert node.get_output_size() == 1
     assert list(node.get_output_shape(0)) == [3, 2]
@@ -162,10 +164,10 @@ def test_axisset():
 
 def test_select():
     element_type = Type.f32
-    A = Parameter(Type.boolean, Shape([1, 2]))
-    B = Parameter(element_type, Shape([1, 2]))
-    C = Parameter(element_type, Shape([1, 2]))
-    node = ov.select(A, B, C)
+    param1 = Parameter(Type.boolean, Shape([1, 2]))
+    param2 = Parameter(element_type, Shape([1, 2]))
+    param3 = Parameter(element_type, Shape([1, 2]))
+    node = ov.select(param1, param2, param3)
     assert node.get_type_name() == "Select"
     assert node.get_output_size() == 1
     assert list(node.get_output_shape(0)) == [1, 2]
@@ -175,7 +177,7 @@ def test_select():
 def test_max_pool_1d():
     element_type = Type.f32
     shape = Shape([1, 1, 10])
-    A = Parameter(element_type, shape)
+    param1 = Parameter(element_type, shape)
     window_shape = [3]
 
     strides = [1] * len(window_shape)
@@ -187,7 +189,7 @@ def test_max_pool_1d():
     idx_elem_type = "i32"
 
     model = ov.max_pool(
-        A,
+        param1,
         strides,
         dilations,
         pads_begin,
@@ -204,10 +206,11 @@ def test_max_pool_1d():
     assert model.get_output_element_type(0) == element_type
     assert model.get_output_element_type(1) == Type.i32
 
+
 def test_max_pool_1d_with_strides():
     element_type = Type.f32
     shape = Shape([1, 1, 10])
-    A = Parameter(element_type, shape)
+    param1 = Parameter(element_type, shape)
     window_shape = [3]
     strides = [2]
     pads_begin = [0] * len(window_shape)
@@ -218,7 +221,7 @@ def test_max_pool_1d_with_strides():
     idx_elem_type = "i32"
 
     model = ov.max_pool(
-        A,
+        param1,
         strides,
         dilations,
         pads_begin,
@@ -236,10 +239,11 @@ def test_max_pool_1d_with_strides():
     assert model.get_output_element_type(0) == element_type
     assert model.get_output_element_type(1) == Type.i32
 
+
 def test_max_pool_2d():
     element_type = Type.f32
     shape = Shape([1, 1, 10, 10])
-    A = Parameter(element_type, shape)
+    param1 = Parameter(element_type, shape)
     window_shape = [3, 3]
     rounding_type = "floor"
     auto_pad = "explicit"
@@ -251,7 +255,7 @@ def test_max_pool_2d():
     pads_end = [0, 0]
 
     model = ov.max_pool(
-        A,
+        param1,
         strides,
         dilations,
         pads_begin,
@@ -272,7 +276,7 @@ def test_max_pool_2d():
 def test_max_pool_2d_with_strides():
     element_type = Type.f32
     shape = Shape([1, 1, 10, 10])
-    A = Parameter(element_type, shape)
+    param1 = Parameter(element_type, shape)
     strides = [2, 2]
     dilations = [1, 1]
     pads_begin = [0, 0]
@@ -283,7 +287,7 @@ def test_max_pool_2d_with_strides():
     idx_elem_type = "i32"
 
     model = ov.max_pool(
-        A,
+        param1,
         strides,
         dilations,
         pads_begin,
diff --git a/src/bindings/python/tests/test_graph/test_ops_result.py b/src/bindings/python/tests/test_graph/test_ops_result.py
new file mode 100644
index 00000000000..105ce81849a
--- /dev/null
+++ b/src/bindings/python/tests/test_graph/test_ops_result.py
@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Copyright (C) 2018-2023 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+import numpy as np
+
+from openvino import PartialShape, Model, Type
+import openvino.runtime.opset13 as ops
+from openvino.runtime.op import Result
+
+
+def test_result():
+    param = ops.parameter(PartialShape([1]), dtype=np.float32, name="param")
+    relu1 = ops.relu(param, name="relu1")
+    result = Result(relu1.output(0))
+    assert result.get_output_element_type(0) == Type.f32
+    assert result.get_output_partial_shape(0) == PartialShape([1])
+    model = Model([result], [param], "test_model")
+
+    result2 = ops.result(relu1, "res2")
+    model.add_results([result2])
+
+    results = model.get_results()
+    assert len(results) == 2
+    assert results[1].get_output_element_type(0) == Type.f32
+    assert results[1].get_output_partial_shape(0) == PartialShape([1])
+    model.remove_result(result)
+    assert len(model.results) == 1
diff --git a/src/bindings/python/tests/test_graph/test_pad.py b/src/bindings/python/tests/test_graph/test_pad.py
index e5d993e59f3..1a6b96c0c96 100644
--- a/src/bindings/python/tests/test_graph/test_pad.py
+++ b/src/bindings/python/tests/test_graph/test_pad.py
@@ -1,24 +1,18 @@
+# -*- coding: utf-8 -*-
 # Copyright (C) 2018-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-# flake8: noqa
 
 import numpy as np
 import pytest
 
 import openvino.runtime.opset13 as ov
-from openvino import  Type
+from openvino import Type
 
 
-@pytest.mark.parametrize(
-        "pad_mode",
-        [
-            "constant",
-            "edge",
-            "reflect",
-            "symmetric",
-        ]
-)
+@pytest.mark.parametrize("pad_mode", [
+    "constant", "edge", "reflect", "symmetric",
+])
 def test_pad_mode(pad_mode):
     pads_begin = np.array([0, 1], dtype=np.int32)
     pads_end = np.array([2, 3], dtype=np.int32)
@@ -32,13 +26,10 @@ def test_pad_mode(pad_mode):
     assert model.get_output_element_type(0) == Type.i32
 
 
-@pytest.mark.parametrize(
-        ("pads_begin", "pads_end", "output_shape"),
-        [
-            ([-1, -1], [-1, -1], [1, 2]),
-            ([2, -1], [-1, 3], [4, 6]),
-        ]
-)
+@pytest.mark.parametrize(("pads_begin", "pads_end", "output_shape"), [
+    ([-1, -1], [-1, -1], [1, 2]),
+    ([2, -1], [-1, 3], [4, 6]),
+])
 def test_pad_being_and_end(pads_begin, pads_end, output_shape):
     input_param = ov.parameter((3, 4), name="input", dtype=np.int32)
     model = ov.pad(input_param, pads_begin, pads_end, "constant")