aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNikhil Raj <nikhil.raj@arm.com>2022-01-04 15:40:11 +0000
committerNikhil Raj <nikhil.raj@arm.com>2022-01-05 12:01:58 +0000
commitfd9de031d9ae1910cf130ef885f3736473964073 (patch)
tree71c837de1ba092ca75d1a7e2fc58dc937d462860
parentcbfd718464b8ac41f0338ae6565d8213d24c0a2a (diff)
downloadarmnn-fd9de031d9ae1910cf130ef885f3736473964073.tar.gz
IVGCVSW-6671 Add CanBeImported to ITensorHandle
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com> Change-Id: If32049f6c101871eb7c94d8e9d9a4688f45ace5a
-rw-r--r--include/armnn/backends/ITensorHandle.hpp11
1 files changed, 11 insertions, 0 deletions
diff --git a/include/armnn/backends/ITensorHandle.hpp b/include/armnn/backends/ITensorHandle.hpp
index d07909972c..78404d0e39 100644
--- a/include/armnn/backends/ITensorHandle.hpp
+++ b/include/armnn/backends/ITensorHandle.hpp
@@ -77,6 +77,17 @@ public:
IgnoreUnused(memory, source);
return false;
};
+
+ /// Implementations must determine if this memory block can be imported.
+ /// This might be based on alignment or memory source type.
+ /// \return true if this memory can be imported.
+ /// \return false by default, cannot be imported.
+ virtual bool CanBeImported(void* memory, MemorySource source)
+ {
+ IgnoreUnused(memory, source);
+ return false;
+ };
+
/// Unimport externally allocated memory
virtual void Unimport()
{};