diff options
Diffstat (limited to 'src/backends/backendsCommon')
-rw-r--r-- | src/backends/backendsCommon/IBackendInternal.hpp | 5 |
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>; |