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)];
Copyright (c) 2020 ARM Limited.
DataType & Get(unsigned int b, unsigned int c, unsigned int h, unsigned int w) const
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout...
unsigned int GetIndex(const armnn::TensorShape &shape, unsigned int batchIndex, unsigned int channelIndex, unsigned int heightIndex, unsigned int widthIndex) const
unsigned int GetNumDimensions() const
TensorBufferArrayView(const TensorShape &shape, DataType *data, armnnUtils::DataLayoutIndexed dataLayout=DataLayout::NCHW)