Skip to content

Commit

Permalink
[TF FE] Refine tests for complex tensors support (openvinotoolkit#20905)
Browse files Browse the repository at this point in the history
Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
  • Loading branch information
rkazants authored Nov 7, 2023
1 parent 5cd9659 commit 494a9cf
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
2 changes: 1 addition & 1 deletion tests/layer_tests/tensorflow_tests/test_tf_ComplexFFT.py
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,7 @@ def create_complex_irfft_net(self, input_shape, fft_length, irfft_op):
dict(input_shape=[1, 3, 20, 40], fft_length=[20, 10], irfft_op=tf.raw_ops.IRFFT2D),
dict(input_shape=[1, 3, 20, 40], fft_length=[10, 40], irfft_op=tf.raw_ops.IRFFT2D),
pytest.param(dict(input_shape=[1, 10, 20, 30, 5], fft_length=[2, 3, 4], irfft_op=tf.raw_ops.IRFFT3D),
marks=pytest.mark.xfail(reason="accuracy-issue-TBD"))
marks=pytest.mark.xfail(reason="accuracy-issue-124452"))
]

@pytest.mark.parametrize("params", test_data_basic)
Expand Down
2 changes: 2 additions & 0 deletions tests/model_hub_tests/tf_hub_tests/precommit_models
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@ imagenet/mobilenet_v1_100_224/classification,https://tfhub.dev/google/imagenet/m
magenta/arbitrary-image-stylization-v1-256,https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2?tf-hub-format=compressed
small_bert/bert_en_uncased_L-4_H-256_A-4,https://tfhub.dev/tensorflow/small_bert/bert_en_uncased_L-4_H-256_A-4/2?tf-hub-format=compressed
movinet/a5/base/kinetics-600/classification,https://tfhub.dev/tensorflow/movinet/a5/base/kinetics-600/classification/3?tf-hub-format=compressed
# model with complex tensors and FFT ops
yamnet,https://tfhub.dev/google/yamnet/1?tf-hub-format=compressed,skip,124451
# secure notebook models
unet/industrial/class_1,https://tfhub.dev/nvidia/unet/industrial/class_1/1?tf-hub-format=compressed
movenet/singlepose/thunder,https://tfhub.dev/google/movenet/singlepose/thunder/4?tf-hub-format=compressed
Expand Down

0 comments on commit 494a9cf

Please sign in to comment.