diff options
Diffstat (limited to 'src/backends/neon/NeonBackend.cpp')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index 01cc6d8119..31e08ceaf5 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -5,6 +5,7 @@ #include "NeonBackend.hpp" #include "NeonBackendId.hpp" +#include "NeonBackendModelContext.hpp" #include "NeonWorkloadFactory.hpp" #include "NeonLayerSupport.hpp" #include "NeonTensorHandleFactory.hpp" @@ -75,9 +76,27 @@ IBackendInternal::Optimizations NeonBackend::GetOptimizations() const return Optimizations{}; } +IBackendInternal::IBackendSpecificModelContextPtr NeonBackend::CreateBackendSpecificModelContext( + const ModelOptions& modelOptions) const +{ + return IBackendSpecificModelContextPtr{new NeonBackendModelContext{modelOptions}}; +} + IBackendInternal::ILayerSupportSharedPtr NeonBackend::GetLayerSupport() const { - static ILayerSupportSharedPtr layerSupport{new NeonLayerSupport}; + static ILayerSupportSharedPtr layerSupport + { + new NeonLayerSupport(IBackendInternal::IBackendSpecificModelContextPtr{}) + }; + return layerSupport; +} + +IBackendInternal::ILayerSupportSharedPtr NeonBackend::GetLayerSupport(const ModelOptions& modelOptions) const +{ + static ILayerSupportSharedPtr layerSupport + { + new NeonLayerSupport(CreateBackendSpecificModelContext(modelOptions)) + }; return layerSupport; } |