diff options
author | Pablo Tello <pablo.tello@arm.com> | 2021-09-01 15:36:23 +0100 |
---|---|---|
committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2021-09-02 08:54:33 +0000 |
commit | 5f3dc253c26e1c839be70672832323158aa4cee0 (patch) | |
tree | 0bd75a32591bd8788a154ce9f2103f5668218aec /include/armnn | |
parent | 31dce2b3fa19781836a9a295b514b2ab37f5d928 (diff) | |
download | armnn-5f3dc253c26e1c839be70672832323158aa4cee0.tar.gz |
MLCE-401 Added helpers to query backend type
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Change-Id: I6c3aa179eeaefa298e5c3a08f4d35708661615f0
Diffstat (limited to 'include/armnn')
-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; } |