aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon')
-rw-r--r--src/backends/neon/NeonTensorHandleFactory.cpp5
-rw-r--r--src/backends/neon/NeonTensorHandleFactory.hpp2
-rw-r--r--src/backends/neon/test/NeonTensorHandleTests.cpp9
3 files changed, 16 insertions, 0 deletions
diff --git a/src/backends/neon/NeonTensorHandleFactory.cpp b/src/backends/neon/NeonTensorHandleFactory.cpp
index ae6ab59efd..1dd83950cd 100644
--- a/src/backends/neon/NeonTensorHandleFactory.cpp
+++ b/src/backends/neon/NeonTensorHandleFactory.cpp
@@ -96,6 +96,11 @@ const FactoryId& NeonTensorHandleFactory::GetId() const
return GetIdStatic();
}
+bool NeonTensorHandleFactory::SupportsInPlaceComputation() const
+{
+ return true;
+}
+
bool NeonTensorHandleFactory::SupportsSubTensors() const
{
return true;
diff --git a/src/backends/neon/NeonTensorHandleFactory.hpp b/src/backends/neon/NeonTensorHandleFactory.hpp
index 41875c988d..954a71451b 100644
--- a/src/backends/neon/NeonTensorHandleFactory.hpp
+++ b/src/backends/neon/NeonTensorHandleFactory.hpp
@@ -61,6 +61,8 @@ public:
const FactoryId& GetId() const override;
+ bool SupportsInPlaceComputation() const override;
+
bool SupportsSubTensors() const override;
MemorySourceFlags GetExportFlags() const override;
diff --git a/src/backends/neon/test/NeonTensorHandleTests.cpp b/src/backends/neon/test/NeonTensorHandleTests.cpp
index 97c7dd3474..3cea29323a 100644
--- a/src/backends/neon/test/NeonTensorHandleTests.cpp
+++ b/src/backends/neon/test/NeonTensorHandleTests.cpp
@@ -697,4 +697,13 @@ BOOST_AUTO_TEST_CASE(NeonTensorHandleFactoryImport)
memoryManager->Release();
}
+BOOST_AUTO_TEST_CASE(NeonTensorHandleSupportsInPlaceComputation)
+{
+ std::shared_ptr<NeonMemoryManager> memoryManager = std::make_shared<NeonMemoryManager>();
+ NeonTensorHandleFactory handleFactory(memoryManager);
+
+ // NeonTensorHandleFactory supports InPlaceComputation
+ ARMNN_ASSERT(handleFactory.SupportsInPlaceComputation());
+}
+
BOOST_AUTO_TEST_SUITE_END()