From 5e082ab852a479198486264dcfaee8c316d015e8 Mon Sep 17 00:00:00 2001
From: YashasSamaga <yashas_2010@yahoo.com>
Date: Mon, 24 Feb 2020 10:45:59 +0530
Subject: [PATCH] resize tensor on reuse if required

---
 modules/dnn/src/op_cuda.hpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/dnn/src/op_cuda.hpp b/modules/dnn/src/op_cuda.hpp
index 44e4955c50..880ed71634 100644
--- 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) {