diff options
Diffstat (limited to 'include/armnn/BackendId.hpp')
| -rw-r--r-- | include/armnn/BackendId.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnn/BackendId.hpp b/include/armnn/BackendId.hpp index 129cbb5d46..89725c165a 100644 --- a/include/armnn/BackendId.hpp +++ b/include/armnn/BackendId.hpp @@ -125,6 +125,8 @@ public: return m_Id < other.m_Id; } + bool IsCpuRef() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuRef); } + const std::string& Get() const { return m_Id; } private: |