aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/TensorBufferArrayView.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/TensorBufferArrayView.hpp')
-rw-r--r--src/backends/reference/workloads/TensorBufferArrayView.hpp13
1 files changed, 7 insertions, 6 deletions
diff --git a/src/backends/reference/workloads/TensorBufferArrayView.hpp b/src/backends/reference/workloads/TensorBufferArrayView.hpp
index 5593ba6a5e..aecec6757a 100644
--- a/src/backends/reference/workloads/TensorBufferArrayView.hpp
+++ b/src/backends/reference/workloads/TensorBufferArrayView.hpp
@@ -7,10 +7,10 @@
#include <armnn/Tensor.hpp>
-#include <backendsCommon/DataLayoutIndexed.hpp>
-
#include <boost/assert.hpp>
+#include <DataLayoutIndexed.hpp>
+
namespace armnn
{
@@ -19,7 +19,8 @@ template <typename DataType>
class TensorBufferArrayView
{
public:
- TensorBufferArrayView(const TensorShape& shape, DataType* data, DataLayoutIndexed dataLayout = DataLayout::NCHW)
+ TensorBufferArrayView(const TensorShape& shape, DataType* data,
+ armnnUtils::DataLayoutIndexed dataLayout = DataLayout::NCHW)
: m_Shape(shape)
, m_Data(data)
, m_DataLayout(dataLayout)
@@ -60,9 +61,9 @@ public:
}
private:
- const TensorShape m_Shape;
- DataType* m_Data;
- DataLayoutIndexed m_DataLayout;
+ const TensorShape m_Shape;
+ DataType* m_Data;
+ armnnUtils::DataLayoutIndexed m_DataLayout;
};
} //namespace armnn