IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ClampFusion for case when min is greater then max (#21509)

Browse Source
This commit is contained in:
Maxim Vafin
2023-12-07 14:58:38 +01:00
committed by GitHub
parent ec40652cea
commit dd8f4f3661
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/layer_tests/pytorch_tests/test_clamp.py
View File

@@ -42,7 +42,7 @@ class TestClamp(PytorchLayerTest):
return aten_clamp(minimum, maximum, as_tensors, op_type), ref_net, op_name
@pytest.mark.parametrize("minimum,maximum",
[(0., 1.), (-0.5, 1.5), (None, 10.), (None, -10.), (10., None), (-10., None), (100, 200)])
[(0., 1.), (-0.5, 1.5), (None, 10.), (None, -10.), (10., None), (-10., None), (100, 200), (1.0, 0.0)])
@pytest.mark.parametrize("as_tensors", [True, False])
@pytest.mark.parametrize("op_type", ["clamp", "clamp_"])
@pytest.mark.nightly
@@ -50,11 +50,6 @@ class TestClamp(PytorchLayerTest):
self._test(*self.create_model(minimum, maximum, as_tensors,
op_type), ie_device, precision, ir_version)
@pytest.mark.xfail(reason='OpenVINO clamp does not support min > max')
def test_clamp_min_greater(self, ie_device, precision, ir_version):
self._test(*self.create_model(1.0, 0.0),
ie_device, precision, ir_version)
class TestClampMin(PytorchLayerTest):
def _prepare_input(self):