From d8014139825d2383cad96caef80653fb3729b39c Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 4 Jun 2024 09:34:48 +0200
Subject: [PATCH] Fix issue eclipse/aidge/aidge#112

---
 include/aidge/backend/opencv/data/TensorImpl.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/include/aidge/backend/opencv/data/TensorImpl.hpp b/include/aidge/backend/opencv/data/TensorImpl.hpp
index c4bc07d..2765c94 100644
--- a/include/aidge/backend/opencv/data/TensorImpl.hpp
+++ b/include/aidge/backend/opencv/data/TensorImpl.hpp
@@ -90,6 +90,8 @@ public:
     // native interface
     const cv::Mat & data() const override { return mData; }
 
+    inline std::size_t capacity() const noexcept override { return (mData.total() * mData.channels()); }
+
     inline std::size_t scalarSize() const noexcept override final { return sizeof(T); }
 
     void copy(const void *src, NbElts_t length, NbElts_t offset = 0) override final {
-- 
GitLab