aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/IBackendInternal.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/IBackendInternal.hpp')
-rw-r--r--src/backends/backendsCommon/IBackendInternal.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/IBackendInternal.hpp b/src/backends/backendsCommon/IBackendInternal.hpp
index fede366475..9c54b821e7 100644
--- a/src/backends/backendsCommon/IBackendInternal.hpp
+++ b/src/backends/backendsCommon/IBackendInternal.hpp
@@ -6,11 +6,13 @@
#include <armnn/Types.hpp>
#include <armnn/IRuntime.hpp>
+#include <vector>
namespace armnn
{
class IWorkloadFactory;
class IBackendContext;
+class Optimization;
class IBackendInternal : public IBackend
{
@@ -26,9 +28,12 @@ public:
using IWorkloadFactoryPtr = std::unique_ptr<IWorkloadFactory>;
using IBackendContextPtr = std::unique_ptr<IBackendContext>;
+ using OptimizationPtr = std::unique_ptr<Optimization>;
+ using Optimizations = std::vector<OptimizationPtr>;
virtual IWorkloadFactoryPtr CreateWorkloadFactory() const = 0;
virtual IBackendContextPtr CreateBackendContext(const IRuntime::CreationOptions&) const = 0;
+ virtual Optimizations GetOptimizations() const = 0;
};
using IBackendInternalUniquePtr = std::unique_ptr<IBackendInternal>;