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

Signed-off-by: Ting Fu <ting.fu@intel.com>
Signed-off-by: Guo Yejun <yejun.guo@intel.com>
This commit is contained in:
Ting Fu 2020-06-18 17:15:36 +08:00 committed by Guo, Yejun
parent 13f5613e68
commit 24d1781cbd

View File

@ -42,6 +42,8 @@ static float get_expected(float f, DNNMathUnaryOperation op)
return asin(f);
case DMUO_ACOS:
return acos(f);
case DMUO_ATAN:
return atan(f);
default:
av_assert0(!"not supported yet");
return 0.f;
@ -97,5 +99,7 @@ int main(int agrc, char **argv)
return 1;
if (test(DMUO_ACOS))
return 1;
if (test(DMUO_ATAN))
return 1;
return 0;
}