diff options
author | David Monahan <david.monahan@arm.com> | 2019-10-02 09:33:57 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2019-10-04 10:16:33 +0000 |
commit | c6e5a6e9f146ecb95704d6fa80fae8465241f09e (patch) | |
tree | f70275018693b8530c6e8338839fad73ca3271da /src/backends/backendsCommon/ITensorHandleFactory.hpp | |
parent | 0ec068f220daf03049a9ffb5ac53118162f50106 (diff) | |
download | armnn-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.hpp | 23 |
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; |