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

[TF FE] Activate TopK layer test with the second output in the pre-commit (#16691)

Browse Source 
* [TF FE] Test the second output for TopK operation

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

* Switch off no sorted case

---------

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev authored and GitHub committed 2023-04-03 11:38:27 +04:00
1 parent b2e4857a64
commit f4fca2d578
1 file changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files
tests/layer_tests/tensorflow_tests/test_tf_TopKV2.py
+3 -3
View File
@@ -36,9 +36,9 @@ class TestTopKV2(CommonTFLayerTest):
dict(input_shape=[10], input_type=tf.float32, k=5, sorted=True, is_first_output=True, is_second_output=False),
dict(input_shape=[2, 3, 10], input_type=tf.int32, k=10, sorted=True, is_first_output=True,
is_second_output=False),
# Expect stable mode support by the CPU plugin. See 101503
pytest.param(dict(input_shape=[4, 12], input_type=tf.float32, k=10, sorted=True, is_first_output=True,
is_second_output=True), marks=pytest.mark.xfail(reason="101503")),
dict(input_shape=[4, 12], input_type=tf.float32, k=10, sorted=True, is_first_output=True,
is_second_output=True),
# Expect stable mode implementation for sort_type=indices in OpenVINO. See 101503
pytest.param(dict(input_shape=[5, 10], input_type=tf.int32, k=8, sorted=False, is_first_output=True,
is_second_output=True), marks=pytest.mark.xfail(reason="101503")),
]