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

[TF FE] Refine tests for complex tensors support (#20905) 

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2023-11-07 10:29:45 +04:00 committed by GitHub
parent 5cd9659033
commit 494a9cf9a9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -182,7 +182,7 @@ class TestComplexIRFFT(CommonTFLayerTest):
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)

2
tests/model_hub_tests/tf_hub_tests/precommit_models
View File

@ -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