diff mbox series

[FFmpeg-devel,V2,12/12] dnn-layer-math-unary-test: add unit test for atanh

Message ID 20200629145411.24005-12-ting.fu@intel.com
State New
Headers show
Series [FFmpeg-devel,V2,01/12] dnn_backend_native_layer_mathunary: add sinh support
Related show

Checks

Context Check Description
andriy/default pending
andriy/make success Make finished
andriy/make_fate success Make fate finished

Commit Message

Ting Fu June 29, 2020, 2:54 p.m. UTC
Signed-off-by: Ting Fu <ting.fu@intel.com>
---
 tests/dnn/dnn-layer-mathunary-test.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)
diff mbox series

Patch

diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c
index 5587e47ad5..683e623d95 100644
--- a/tests/dnn/dnn-layer-mathunary-test.c
+++ b/tests/dnn/dnn-layer-mathunary-test.c
@@ -54,6 +54,8 @@  static float get_expected(float f, DNNMathUnaryOperation op)
         return asinh(f);
     case DMUO_ACOSH:
         return acosh(f);
+    case DMUO_ATANH:
+        return atanh(f);
     default:
         av_assert0(!"not supported yet");
         return 0.f;
@@ -65,8 +67,8 @@  static int test(DNNMathUnaryOperation op)
     DnnLayerMathUnaryParams params;
     DnnOperand operands[2];
     int32_t input_indexes[1];
-    float input[1*1*2*3] = {
-        -3, 2.5, 2, -2.1, 7.8, 100};
+    float input[1*1*3*3] = {
+        0.1, 0.5, 0.75, -3, 2.5, 2, -2.1, 7.8, 100};
     float *output;
 
     params.un_op = op;
@@ -74,7 +76,7 @@  static int test(DNNMathUnaryOperation op)
     operands[0].data = input;
     operands[0].dims[0] = 1;
     operands[0].dims[1] = 1;
-    operands[0].dims[2] = 2;
+    operands[0].dims[2] = 3;
     operands[0].dims[3] = 3;
     operands[1].data = NULL;
 
@@ -121,5 +123,7 @@  int main(int agrc, char **argv)
         return 1;
     if (test(DMUO_ACOSH))
         return 1;
+    if (test(DMUO_ATANH))
+        return 1;
     return 0;
 }