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

[TF FE][Tests] Add layer tests for SegmentSum and DynamicPartition in the pre-commit (#13812)

Browse Source 
After fixes for TopK and arithmetic operations for integer tensors, we are ready to switch
these tests in the pre-commit

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2022-11-04 05:02:48 +03:00 committed by GitHub
parent 7a6960b2f4
commit 954b9d3149
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/layer_tests/tensorflow_tests/test_tf_DynamicPartition.py
View File

@ -42,7 +42,7 @@ class TestDynamicPartition(CommonTFLayerTest):
]
@pytest.mark.parametrize("params", test_data_basic)
@pytest.mark.nightly
@pytest.mark.precommit_tf_fe
def test_dynamic_partition_basic(self, params, ie_device, precision, ir_version, temp_dir,
use_new_frontend, use_old_api):
if not use_new_frontend:

2
tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py
View File

@ -88,4 +88,4 @@ class TestParallelDynamicStitch(CommonTFLayerTest):
pytest.skip("DynamicStitch operation is not supported via legacy frontend.")
self._test(*self.create_parallel_dynamic_stitch_net(**params),
ie_device, precision, ir_version, temp_dir=temp_dir,
use_new_frontend=use_new_frontend, use_old_api=use_old_api)
use_new_frontend=use_new_frontend, use_old_api=use_old_api)

2
tests/layer_tests/tensorflow_tests/test_tf_SegmentSum.py
View File

@ -39,7 +39,7 @@ class TestSegmentSum(CommonTFLayerTest):
]
@pytest.mark.parametrize("params", test_data_basic)
@pytest.mark.nightly
@pytest.mark.precommit_tf_fe
def test_segment_sum_basic(self, params, ie_device, precision, ir_version, temp_dir,
use_new_frontend, use_old_api):
if not use_new_frontend: