diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-22 16:10:44 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-22 16:10:44 +0000 |
commit | 649dd9515ddf4bd00a0bff64d51dfd835a6c7b39 (patch) | |
tree | c938bc8eb11dd24223c0cb00a57d4372a907b943 /src/backends/backendsCommon/CpuTensorHandle.hpp | |
parent | 382e21ce95c04479a6900afca81a57949b369f1e (diff) | |
download | armnn-649dd9515ddf4bd00a0bff64d51dfd835a6c7b39.tar.gz |
IVGCVSW-2467 Remove GetDataType<T> function
Change-Id: I7359617a307b9abb4c30b3d5f2364dc6d0f828f0
Diffstat (limited to 'src/backends/backendsCommon/CpuTensorHandle.hpp')
-rw-r--r-- | src/backends/backendsCommon/CpuTensorHandle.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/CpuTensorHandle.hpp b/src/backends/backendsCommon/CpuTensorHandle.hpp index b88a0d385b..dd6413f2e7 100644 --- a/src/backends/backendsCommon/CpuTensorHandle.hpp +++ b/src/backends/backendsCommon/CpuTensorHandle.hpp @@ -5,6 +5,7 @@ #pragma once #include "CpuTensorHandleFwd.hpp" +#include "CompatibleTypes.hpp" #include <armnn/TypesUtils.hpp> @@ -22,7 +23,7 @@ public: template <typename T> const T* GetConstTensor() const { - BOOST_ASSERT(GetTensorInfo().GetDataType() == GetDataType<T>()); + BOOST_ASSERT(CompatibleTypes<T>(GetTensorInfo().GetDataType())); return reinterpret_cast<const T*>(m_Memory); } @@ -82,7 +83,7 @@ public: template <typename T> T* GetTensor() const { - BOOST_ASSERT(GetTensorInfo().GetDataType() == GetDataType<T>()); + BOOST_ASSERT(CompatibleTypes<T>(GetTensorInfo().GetDataType())); return reinterpret_cast<T*>(m_MutableMemory); } |