dnn_backend_native_layer_mathbinary: add floormod support

Signed-off-by: Mingyu Yin <mingyu.yin@intel.com>

2 files changed, 8 insertions, 0 deletions

diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
index 38742db7a6..7d81694288 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
+++ b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
@@ -50,6 +50,10 @@ static float minimum(float src0, float src1)
 {
     return FFMIN(src0, src1);
 }
+static float floormod(float src0, float src1)
+{
+    return (float)((int)(src0) % (int)(src1));
+}
 
 static void math_binary_commutative(FunType pfun, const DnnLayerMathBinaryParams *params, const DnnOperand *input, DnnOperand *output, DnnOperand *operands, const int32_t *input_operand_indexes)
 {
@@ -178,6 +182,9 @@ int dnn_execute_layer_math_binary(DnnOperand *operands, const int32_t *input_ope
     case DMBO_MINIMUM:
         math_binary_commutative(minimum, params, input, output, operands, input_operand_indexes);
         return 0;
+    case DMBO_FLOORMOD:
+        math_binary_not_commutative(floormod, params, input, output, operands, input_operand_indexes);
+        return 0;
     default:
         return -1;
     }
diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.h b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.h
index 0acf3b0ea0..9525685afa 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.h
+++ b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.h
@@ -36,6 +36,7 @@ typedef enum {
     DMBO_MUL = 2,
     DMBO_REALDIV = 3,
     DMBO_MINIMUM = 4,
+    DMBO_FLOORMOD = 5,
     DMBO_COUNT
 } DNNMathBinaryOperation;