diff options
Diffstat (limited to 'src/backends/backendsCommon/test')
-rw-r--r-- | src/backends/backendsCommon/test/MockBackend.cpp | 2 | ||||
-rw-r--r-- | src/backends/backendsCommon/test/MockBackend.hpp | 25 |
2 files changed, 16 insertions, 11 deletions
diff --git a/src/backends/backendsCommon/test/MockBackend.cpp b/src/backends/backendsCommon/test/MockBackend.cpp index d8f8cfe701..ac99738a92 100644 --- a/src/backends/backendsCommon/test/MockBackend.cpp +++ b/src/backends/backendsCommon/test/MockBackend.cpp @@ -99,7 +99,7 @@ IBackendInternal::IBackendContextPtr MockBackend::CreateBackendContext(const IRu } IBackendInternal::IBackendProfilingContextPtr MockBackend::CreateBackendProfilingContext( - const IRuntime::CreationOptions&, armnn::profiling::IBackendProfiling&) const + const IRuntime::CreationOptions&, IBackendProfilingPtr&) { return IBackendProfilingContextPtr{}; } diff --git a/src/backends/backendsCommon/test/MockBackend.hpp b/src/backends/backendsCommon/test/MockBackend.hpp index 3fe3100986..d1a0082e2e 100644 --- a/src/backends/backendsCommon/test/MockBackend.hpp +++ b/src/backends/backendsCommon/test/MockBackend.hpp @@ -5,14 +5,15 @@ #pragma once +#include <LayerSupportCommon.hpp> #include <armnn/backends/IBackendInternal.hpp> #include <armnn/backends/OptimizationViews.hpp> -#include <LayerSupportCommon.hpp> #include <backendsCommon/LayerSupportBase.hpp> namespace armnn { + class MockBackend : public IBackendInternal { public: @@ -20,17 +21,20 @@ public: ~MockBackend() = default; static const BackendId& GetIdStatic(); - const BackendId& GetId() const override { return GetIdStatic(); } + const BackendId& GetId() const override + { + return GetIdStatic(); + } IBackendInternal::IMemoryManagerUniquePtr CreateMemoryManager() const override; - IBackendInternal::IWorkloadFactoryPtr CreateWorkloadFactory( - const IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr) const override; + IBackendInternal::IWorkloadFactoryPtr + CreateWorkloadFactory(const IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr) const override; IBackendInternal::IBackendContextPtr CreateBackendContext(const IRuntime::CreationOptions&) const override; - IBackendInternal::IBackendProfilingContextPtr CreateBackendProfilingContext( - const IRuntime::CreationOptions& creationOptions, - armnn::profiling::IBackendProfiling& backendProfiling) const override; + IBackendInternal::IBackendProfilingContextPtr + CreateBackendProfilingContext(const IRuntime::CreationOptions& creationOptions, + IBackendProfilingPtr& backendProfiling) override; IBackendInternal::Optimizations GetOptimizations() const override; IBackendInternal::ILayerSupportSharedPtr GetLayerSupport() const override; @@ -38,7 +42,8 @@ public: OptimizationViews OptimizeSubgraphView(const SubgraphView& subgraph) const override; }; -class MockLayerSupport : public LayerSupportBase { +class MockLayerSupport : public LayerSupportBase +{ public: bool IsInputSupported(const TensorInfo& /*input*/, Optional<std::string&> /*reasonIfUnsupported = EmptyOptional()*/) const override @@ -47,7 +52,7 @@ public: } bool IsOutputSupported(const TensorInfo& /*input*/, - Optional<std::string&> /*reasonIfUnsupported = EmptyOptional()*/) const override + Optional<std::string&> /*reasonIfUnsupported = EmptyOptional()*/) const override { return true; } @@ -71,4 +76,4 @@ public: } }; -} // namespace armnn +} // namespace armnn |