diff options
author | David Beck <david.beck@arm.com> | 2018-10-18 15:13:56 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | 33f0ae0d293f5048089f2a04985a8a8bfa1d75a6 (patch) | |
tree | 19f27463e524150c26ac75a2ea5c8e6c95f954c6 /src/armnn/Layer.hpp | |
parent | ceae3aa1b619161d49fd2847d3c73d6a858b2b8c (diff) | |
download | armnn-33f0ae0d293f5048089f2a04985a8a8bfa1d75a6.tar.gz |
IVGCVSW-2019 : replace Compute enum in the Layer object
Change-Id: I76551d511ef718eac36e5b8e5fe426ec3a402855
Diffstat (limited to 'src/armnn/Layer.hpp')
-rw-r--r-- | src/armnn/Layer.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index d897b255a6..a969607100 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -237,8 +237,8 @@ public: DataLayout GetDataLayout() const { return m_DataLayout; } - Compute GetComputeDevice() const { return m_ComputeDevice; } - void SetComputeDevice(Compute device) { m_ComputeDevice = device; } + const BackendId& GetBackendId() const { return m_BackendId; } + void SetBackendId(const BackendId& id) { m_BackendId = id; } // Virtuals @@ -345,7 +345,7 @@ private: const LayerType m_Type; const DataLayout m_DataLayout; - Compute m_ComputeDevice; + BackendId m_BackendId; /// Used for sorting. mutable LayerPriority m_Priority = 0; |