From 0de50430609147248b8ecacaa06bf957328f7f8e Mon Sep 17 00:00:00 2001 From: Ting Fu Date: Mon, 29 Jun 2020 22:54:05 +0800 Subject: dnn-layer-math-unary-test: add unit test for tanh Signed-off-by: Ting Fu --- tests/dnn/dnn-layer-mathunary-test.c | 4 ++++ 1 file changed, 4 insertions(+) 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; } -- cgit v1.2.3