Merge pull request #24880 from fengyuentau:scatter_fp16_fix

dnn: fix ocl_fp16 support of scatter layer
10 months ago · e4131c0b5b
parent bfad61f433 d269de0a03
commit e4131c0b5b
2 changed files with 10 additions and 0 deletions
--- a/modules/dnn/src/layers/scatterND_layer.cpp
+++ b/modules/dnn/src/layers/scatterND_layer.cpp
@ -74,6 +74,11 @@ public:
        CV_TRACE_FUNCTION();
        CV_TRACE_ARG_VALUE(name, "name", name.c_str());

+        if (inputs_arr.depth() == CV_16S) {
+            forward_fallback(inputs_arr, outputs_arr, internals_arr);
+            return;
+        }
+
        std::vector<Mat> inputs, outputs;
        inputs_arr.getMatVector(inputs);
        outputs_arr.getMatVector(outputs);
--- a/modules/dnn/src/layers/scatter_layer.cpp
+++ b/modules/dnn/src/layers/scatter_layer.cpp
@ -68,6 +68,11 @@ public:
        CV_TRACE_FUNCTION();
        CV_TRACE_ARG_VALUE(name, "name", name.c_str());

+        if (inputs_arr.depth() == CV_16S) {
+            forward_fallback(inputs_arr, outputs_arr, internals_arr);
+            return;
+        }
+
        std::vector<Mat> inputs, outputs;
        inputs_arr.getMatVector(inputs);
        outputs_arr.getMatVector(outputs);