Merge pull request #24427 from fengyuentau:gather_elements_fp16

dnn: fp16 support for gather elements
pull/24429/head
Alexander Smorkalov 1 year ago committed by GitHub
commit a9664abb57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      modules/dnn/src/layers/gather_elements_layer.cpp

@ -70,6 +70,12 @@ 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);

Loading…
Cancel
Save