diff options
| author | Francis Murtagh <francis.murtagh@arm.com> | 2021-10-14 17:30:24 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-10-19 22:26:45 +0100 |
| commit | e8d7ccb1a18b5c39ecf17fb063fcc50baacc74b3 (patch) | |
| tree | d1a072111807145a786b4c4221707bd34c1373be /src/backends/neon/NeonBackend.cpp | |
| parent | fc7c2f3c1dd990620a01ecf1a6641813e6f46ef8 (diff) | |
| download | armnn-e8d7ccb1a18b5c39ecf17fb063fcc50baacc74b3.tar.gz | |
IVGCVSW-6310 Update ICustomAllocator and add GetDefaultAllocator function to backends
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: If2879e8a82692285e7fa3f4d09abd608ca28de12
Diffstat (limited to 'src/backends/neon/NeonBackend.cpp')
| -rw-r--r-- | src/backends/neon/NeonBackend.cpp | 25 |
1 files changed, 16 insertions, 9 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index 2c3abfd70d..0500ee3411 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -22,15 +22,16 @@ #include <armnn/utility/PolymorphicDowncast.hpp> -#include "workloads/NeonAdditionWorkload.hpp" -#include "workloads/NeonBatchNormalizationWorkload.hpp" -#include "workloads/NeonConvolution2dWorkload.hpp" -#include "workloads/NeonDepthwiseConvolutionWorkload.hpp" -#include "workloads/NeonDivisionWorkload.hpp" -#include "workloads/NeonFullyConnectedWorkload.hpp" -#include "workloads/NeonMultiplicationWorkload.hpp" -#include "workloads/NeonReduceWorkload.hpp" -#include "workloads/NeonSubtractionWorkload.hpp" +#include <neon/workloads/NeonAdditionWorkload.hpp> +#include <neon/workloads/NeonBatchNormalizationWorkload.hpp> +#include <neon/workloads/NeonConvolution2dWorkload.hpp> +#include <neon/workloads/NeonDepthwiseConvolutionWorkload.hpp> +#include <neon/workloads/NeonDivisionWorkload.hpp> +#include <neon/workloads/NeonFullyConnectedWorkload.hpp> +#include <neon/workloads/NeonMultiplicationWorkload.hpp> +#include <neon/workloads/NeonReduceWorkload.hpp> +#include <neon/workloads/NeonSubtractionWorkload.hpp> +#include <backendsCommon/DefaultAllocator.hpp> #include <Optimizer.hpp> @@ -427,4 +428,10 @@ void NeonBackend::RegisterTensorHandleFactories(class TensorHandleFactoryRegistr registry.RegisterFactory(std::make_unique<NeonTensorHandleFactory>(memoryManager)); } +std::unique_ptr<ICustomAllocator> NeonBackend::GetDefaultAllocator() const +{ + return std::make_unique<DefaultAllocator>(); +} + + } // namespace armnn |