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

[TF FE] Add layer test for Mish activation function (#16557)

Browse Source 
Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2023-03-28 20:43:11 +04:00 committed by GitHub
parent 44f0419a0b
commit 17c3e67336
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/layer_tests/requirements.txt
View File

@ -3,3 +3,4 @@ numpy>=1.19.2
torch
torchvision
pytest
tensorflow-addons

11
tests/layer_tests/tensorflow_tests/test_tf_UnaryOps.py
View File

@ -54,6 +54,8 @@ class TestUnaryOps(CommonTFLayerTest):
"""
import tensorflow as tf
import tensorflow_addons as tfa
self.current_op_type = op_type
op_type_to_tf = {
'Abs': tf.math.abs,
@ -72,6 +74,7 @@ class TestUnaryOps(CommonTFLayerTest):
'Floor': tf.math.floor,
'Log': tf.math.log,
'LogicalNot': tf.math.logical_not,
'Mish': tfa.activations.mish,
'Negative': tf.math.negative,
'Sigmoid': tf.nn.sigmoid,
'Sign': tf.math.sign,
@ -95,7 +98,11 @@ class TestUnaryOps(CommonTFLayerTest):
tf_x_shape = permute_nchw_to_nhwc(tf_x_shape, use_new_frontend)
input = tf.compat.v1.placeholder(type, tf_x_shape, 'Input')
op_type_to_tf[self.current_op_type](input, name='Operation')
if self.current_op_type == 'Mish':
# Mish has no attribute name
op_type_to_tf[self.current_op_type](input)
else:
op_type_to_tf[self.current_op_type](input, name='Operation')
tf.compat.v1.global_variables_initializer()
tf_net = sess.graph_def
@ -155,6 +162,7 @@ class TestUnaryOps(CommonTFLayerTest):
'LogicalNot',
'Square',
'Erf',
'Mish',
])
@pytest.mark.precommit
def test_unary_op_precommit(self, params, ie_device, precision, ir_version, temp_dir, op_type,
@ -198,6 +206,7 @@ class TestUnaryOps(CommonTFLayerTest):
'Asinh',
'Square',
'Erf',
'Mish',
])
@pytest.mark.nightly
def test_unary_op(self, params, ie_device, precision, ir_version, temp_dir, op_type,