dnn-layer-mathbinary-test: add unit test for 'mul'

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
This commit is contained in:
Guo, Yejun 2020-04-11 13:24:36 +08:00
parent ef79408e97
commit 265b5bd324
1 changed files with 5 additions and 0 deletions

View File

@ -34,6 +34,8 @@ static float get_expected(float f1, float f2, DNNMathBinaryOperation op)
return f1 - f2; return f1 - f2;
case DMBO_ADD: case DMBO_ADD:
return f1 + f2; return f1 + f2;
case DMBO_MUL:
return f1 * f2;
default: default:
av_assert0(!"not supported yet"); av_assert0(!"not supported yet");
return 0.f; return 0.f;
@ -190,5 +192,8 @@ int main(int argc, char **argv)
if (test(DMBO_ADD)) if (test(DMBO_ADD))
return 1; return 1;
if (test(DMBO_MUL))
return 1;
return 0; return 0;
} }