diff options
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefBackend.cpp | 4 | ||||
-rw-r--r-- | src/backends/reference/RefBackend.hpp | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp index 53c55ab26a..85c6a2e1e4 100644 --- a/src/backends/reference/RefBackend.cpp +++ b/src/backends/reference/RefBackend.cpp @@ -71,8 +71,8 @@ IBackendInternal::ILayerSupportSharedPtr RefBackend::GetLayerSupport() const bool RefBackend::HasCapability(BackendCapability capabilityClass) const { - auto search = backendCapabilities.find(capabilityClass); - if (search != backendCapabilities.end()) + auto search = cpuRefCapabilities.find(capabilityClass); + if (search != cpuRefCapabilities.end()) { return true; } diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp index c92936ca0c..3143173061 100644 --- a/src/backends/reference/RefBackend.hpp +++ b/src/backends/reference/RefBackend.hpp @@ -9,7 +9,7 @@ namespace armnn { -const std::set<armnn::BackendCapability> backendCapabilities { +const std::set<armnn::BackendCapability> cpuRefCapabilities { armnn::BackendCapability::NonConstWeights, }; |