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) 2021 ARM Limited and Contributors.
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
#define ARMNN_ASSERT(COND)
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
TensorBufferArrayView(const TensorShape &shape, DataType *data, armnnUtils::DataLayoutIndexed dataLayout=DataLayout::NCHW)