diff options
Diffstat (limited to 'include')
| -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 c94cd66661..5849dea8fb 100644 --- a/include/armnn/BackendId.hpp +++ b/include/armnn/BackendId.hpp @@ -132,6 +132,8 @@ public: } bool IsCpuRef() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuRef); } + bool IsCpuAcc() const { return m_Id == GetComputeDeviceAsCString(Compute::CpuAcc); } + bool IsGpuAcc() const { return m_Id == GetComputeDeviceAsCString(Compute::GpuAcc); } const std::string& Get() const { return m_Id; } |