aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/ClTensorHandle.hpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2021-04-14 16:55:36 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2021-05-04 17:06:52 +0100
commite4a41dc5d5fc0f283c01b3260affdfdf6cfc1895 (patch)
treed0873f60422b46bdbe8ebe1dbef22bc87bf75175 /src/backends/cl/ClTensorHandle.hpp
parentc21025dc3c07d60568dd27d816bcdf0575f7695a (diff)
downloadarmnn-e4a41dc5d5fc0f283c01b3260affdfdf6cfc1895.tar.gz
IVGCVSW-5727 Implement Import function of ClImportTensorHandle
* Split ClImportTensorHandle out from ClTenorHandle * Added implementation of Import function * Added Unit Tests Signed-off-by: David Monahan <david.monahan@arm.com> Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: I07de2ca5bebf19dfb9a8dddea4b18340ffc31fad
Diffstat (limited to 'src/backends/cl/ClTensorHandle.hpp')
-rw-r--r--src/backends/cl/ClTensorHandle.hpp12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/backends/cl/ClTensorHandle.hpp b/src/backends/cl/ClTensorHandle.hpp
index 061117e9a6..5720d2cf11 100644
--- a/src/backends/cl/ClTensorHandle.hpp
+++ b/src/backends/cl/ClTensorHandle.hpp
@@ -57,7 +57,11 @@ public:
virtual void Allocate() override
{
// If we have enabled Importing, don't allocate the tensor
- if (!m_IsImportEnabled)
+ if (m_IsImportEnabled)
+ {
+ throw MemoryImportException("ClTensorHandle::Attempting to allocate memory when importing");
+ }
+ else
{
armnn::armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_Tensor);
}
@@ -67,7 +71,11 @@ public:
virtual void Manage() override
{
// If we have enabled Importing, don't manage the tensor
- if (!m_IsImportEnabled)
+ if (m_IsImportEnabled)
+ {
+ throw MemoryImportException("ClTensorHandle::Attempting to manage memory when importing");
+ }
+ else
{
assert(m_MemoryGroup != nullptr);
m_MemoryGroup->manage(&m_Tensor);