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

Signed-off-by: Ting Fu <ting.fu@intel.com>
This commit is contained in:
Ting Fu 2020-06-29 22:54:11 +08:00 committed by Guo, Yejun
parent c0cdeea0ee
commit 57ea0483af
1 changed files with 7 additions and 3 deletions

View File

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