dnn-layer-mathbinary-test: add unit test for minimum

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
5 years ago · 6fd61234d5
parent 71e28c5422
commit 6fd61234d5
1 changed files with 5 additions and 0 deletions
--- a/tests/dnn/dnn-layer-mathbinary-test.c
+++ b/tests/dnn/dnn-layer-mathbinary-test.c
@ -38,6 +38,8 @@ static float get_expected(float f1, float f2, DNNMathBinaryOperation op)
        return f1 * f2;
    case DMBO_REALDIV:
        return f1 / f2;
+    case DMBO_MINIMUM:
+        return (f1 < f2) ? f1 : f2;
    default:
        av_assert0(!"not supported yet");
        return 0.f;
@ -200,5 +202,8 @@ int main(int argc, char **argv)
    if (test(DMBO_REALDIV))
        return 1;

+    if (test(DMBO_MINIMUM))
+        return 1;
+
    return 0;
 }