aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/backends/IBackendInternal.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/backends/IBackendInternal.hpp')
-rw-r--r--include/armnn/backends/IBackendInternal.hpp20
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>;