Browse Source

dnn-layer-math-unary-test: add unit test for tanh

Signed-off-by: Ting Fu <ting.fu@intel.com>
tags/n4.4
Ting Fu Guo, Yejun 5 years ago
parent
commit
0de5043060
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      tests/dnn/dnn-layer-mathunary-test.c

+ 4
- 0
tests/dnn/dnn-layer-mathunary-test.c View File

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

Loading…
Cancel
Save