diff options
Diffstat (limited to 'include/armnn/backends/IBackendInternal.hpp')
-rw-r--r-- | include/armnn/backends/IBackendInternal.hpp | 20 |
1 files changed, 14 insertions, 6 deletions
diff --git a/include/armnn/backends/IBackendInternal.hpp b/include/armnn/backends/IBackendInternal.hpp index f4fe678a5b..7500e35897 100644 --- a/include/armnn/backends/IBackendInternal.hpp +++ b/include/armnn/backends/IBackendInternal.hpp @@ -13,12 +13,12 @@ #include <SubgraphView.hpp> #include <optimizations/Optimization.hpp> -#include "IBackendContext.hpp" -#include "armnn/backends/profiling/IBackendProfiling.hpp" -#include "armnn/backends/profiling/IBackendProfilingContext.hpp" -#include "IMemoryManager.hpp" -#include "ITensorHandleFactory.hpp" -#include "OptimizationViews.hpp" +#include <armnn/backends/IBackendContext.hpp> +#include <armnn/backends/IMemoryManager.hpp> +#include <armnn/backends/ITensorHandleFactory.hpp> +#include <armnn/backends/OptimizationViews.hpp> +#include <armnn/backends/profiling/IBackendProfiling.hpp> +#include <armnn/backends/profiling/IBackendProfilingContext.hpp> #include <vector> #include <memory> @@ -198,6 +198,14 @@ public: } return false; } + + /// Returns the default memory allocator for the backend + /// + /// \return - Returns unique pointer to the Default Allocator of the Backend + virtual std::unique_ptr<ICustomAllocator> GetDefaultAllocator() const + { + throw armnn::Exception("GetDefaultAllocator: Function has not been implemented in backend."); + } }; using IBackendInternalUniquePtr = std::unique_ptr<IBackendInternal>; |