diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 25 | ||||
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 2 |
2 files changed, 18 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 diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 37e1722984..68d60a4c04 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -65,6 +65,8 @@ public: { return cpuAccCapabilities; }; + + std::unique_ptr<ICustomAllocator> GetDefaultAllocator() const override; }; } // namespace armnn |