virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/dnn/dnn-layer-mathunary-test.c
View File

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