resize tensor on reuse if required

5e082ab8 · YashasSamaga · 8eafddf2 · 5e082ab8
Commit 5e082ab8 authored Feb 24, 2020 by YashasSamaga
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

op_cuda.hpp modules/dnn/src/op_cuda.hpp +4 -0

No files found.
--- a/modules/dnn/src/op_cuda.hpp
+++ b/modules/dnn/src/op_cuda.hpp
@@ -305,6 +305,10 @@ namespace cv { namespace dnn {
            shape = shape_;
            offset = 0;
            shared_block = base->shared_block;
+
+            auto numel = total(shape_);
+            if (numel > shared_block->device.size())
+                shared_block->device.reset(numel);
        }

        static Ptr<BackendWrapper> create(Mat& m) {