diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-08-25 11:48:00 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-08-25 13:02:43 +0000 |
commit | ab3bd4d48e75f75c1729a177b35aff61ed0fcd4e (patch) | |
tree | 8992d946f488ef7d0444ba88712008135b83a375 /src/backends/neon | |
parent | 2b1c155d2eeeb9650d49a7baec9b68b4aca72bb1 (diff) | |
download | armnn-ab3bd4d48e75f75c1729a177b35aff61ed0fcd4e.tar.gz |
IVGCVSW-5109 'Add SupportsInPlaceComputation to TensorHandleFactories'
* Added functionality to query if TensorHandleFactory supports InPlaceComputation
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Icf5bfc5f999fc5d03681dcb8cec88d921842458b
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.cpp | 5 | ||||
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.hpp | 2 | ||||
-rw-r--r-- | src/backends/neon/test/NeonTensorHandleTests.cpp | 9 |
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() |