23.05
|
Go to the documentation of this file.
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 GetIndex(const armnn::TensorShape &shape, unsigned int batchIndex, unsigned int channelIndex, unsigned int heightIndex, unsigned int widthIndex) const
DataType & Get(unsigned int b, unsigned int c, unsigned int h, unsigned int w) const
Copyright (c) 2021 ARM Limited and Contributors.
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
TensorBufferArrayView(const TensorShape &shape, DataType *data, armnnUtils::DataLayoutIndexed dataLayout=DataLayout::NCHW)
#define ARMNN_ASSERT(COND)
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout.