12 #include <boost/assert.hpp> 18 template <
typename DataType>
26 , m_DataLayout(dataLayout)
31 DataType&
Get(
unsigned int b,
unsigned int c,
unsigned int h,
unsigned int w)
const 33 return m_Data[m_DataLayout.
GetIndex(m_Shape, b, c, h, w)];
unsigned int GetNumDimensions() const
DataType & Get(unsigned int b, unsigned int c, unsigned int h, unsigned int w) const
TensorBufferArrayView(const TensorShape &shape, DataType *data, armnnUtils::DataLayoutIndexed dataLayout=DataLayout::NCHW)
unsigned int GetIndex(const armnn::TensorShape &shape, unsigned int batchIndex, unsigned int channelIndex, unsigned int heightIndex, unsigned int widthIndex) const