aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/ITensorHandleFactory.hpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2019-10-02 09:33:57 +0100
committerDavid Monahan <david.monahan@arm.com>2019-10-04 10:16:33 +0000
commitc6e5a6e9f146ecb95704d6fa80fae8465241f09e (patch)
treef70275018693b8530c6e8338839fad73ca3271da /src/backends/backendsCommon/ITensorHandleFactory.hpp
parent0ec068f220daf03049a9ffb5ac53118162f50106 (diff)
downloadarmnn-c6e5a6e9f146ecb95704d6fa80fae8465241f09e.tar.gz
IVGCVSW-3925 Add Backward compatibility for ITensorHandle CreateTensorHandle functions
Change-Id: I940b7ca706c9a8bc38743176eb7959aa629a6876 Signed-off-by: David Monahan <david.monahan@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/ITensorHandleFactory.hpp')
-rw-r--r--src/backends/backendsCommon/ITensorHandleFactory.hpp23
1 files changed, 21 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/ITensorHandleFactory.hpp b/src/backends/backendsCommon/ITensorHandleFactory.hpp
index c6deaef6bb..2e4742301b 100644
--- a/src/backends/backendsCommon/ITensorHandleFactory.hpp
+++ b/src/backends/backendsCommon/ITensorHandleFactory.hpp
@@ -8,6 +8,9 @@
#include <armnn/IRuntime.hpp>
#include <armnn/MemorySources.hpp>
#include <armnn/Types.hpp>
+#include "ITensorHandle.hpp"
+
+#include <boost/core/ignore_unused.hpp>
namespace armnn
{
@@ -25,12 +28,28 @@ public:
TensorShape const& subTensorShape,
unsigned int const* subTensorOrigin) const = 0;
+ virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo) const = 0;
+
+ virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
+ DataLayout dataLayout) const = 0;
+
+ // Utility Functions for backends which require TensorHandles to have unmanaged memory.
+ // These should be overloaded if required to facilitate direct import of input tensors
+ // and direct export of output tensors.
virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
- const bool IsMemoryManaged = true) const = 0;
+ const bool IsMemoryManaged) const
+ {
+ boost::ignore_unused(IsMemoryManaged);
+ return CreateTensorHandle(tensorInfo);
+ }
virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
DataLayout dataLayout,
- const bool IsMemoryManaged = true) const = 0;
+ const bool IsMemoryManaged) const
+ {
+ boost::ignore_unused(IsMemoryManaged);
+ return CreateTensorHandle(tensorInfo, dataLayout);
+ }
virtual const FactoryId& GetId() const = 0;