diff options
Diffstat (limited to 'src/backends/reference/workloads/TensorBufferArrayView.hpp')
-rw-r--r-- | src/backends/reference/workloads/TensorBufferArrayView.hpp | 13 |
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 |