summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTing Fu <ting.fu@intel.com>2020-06-29 22:54:05 +0800
committerGuo, Yejun <yejun.guo@intel.com>2020-07-06 12:45:14 +0800
commit0de50430609147248b8ecacaa06bf957328f7f8e (patch)
treec773f9a141ceb21f3ca5f8c3d50b472f9a7306f1
parentea71e731f413bc581c20c0f963b13e8410dd3b6f (diff)
dnn-layer-math-unary-test: add unit test for tanh
Signed-off-by: Ting Fu <ting.fu@intel.com>
-rw-r--r--tests/dnn/dnn-layer-mathunary-test.c4
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c
index 0280debc0b..6885b4d318 100644
--- a/tests/dnn/dnn-layer-mathunary-test.c
+++ b/tests/dnn/dnn-layer-mathunary-test.c
@@ -48,6 +48,8 @@ static float get_expected(float f, DNNMathUnaryOperation op)
return sinh(f);
case DMUO_COSH:
return cosh(f);
+ case DMUO_TANH:
+ return tanh(f);
default:
av_assert0(!"not supported yet");
return 0.f;
@@ -109,5 +111,7 @@ int main(int agrc, char **argv)
return 1;
if (test(DMUO_COSH))
return 1;
+ if (test(DMUO_TANH))
+ return 1;
return 0;
}