#include <TensorBufferArrayView.hpp>

Public Member Functions
	TensorBufferArrayView (const TensorShape &shape, DataType *data, armnnUtils::DataLayoutIndexed dataLayout=DataLayout::NCHW)

DataType &	Get (unsigned int b, unsigned int c, unsigned int h, unsigned int w) const

Detailed Description

template<typename DataType>
class armnn::TensorBufferArrayView< DataType >

Definition at line 19 of file TensorBufferArrayView.hpp.

Constructor & Destructor Documentation

◆ TensorBufferArrayView()

TensorBufferArrayView	(	const TensorShape &	shape,
		DataType *	data,
		armnnUtils::DataLayoutIndexed	dataLayout = `DataLayout::NCHW`
	)

inline

Definition at line 22 of file TensorBufferArrayView.hpp.

References ARMNN_ASSERT, and TensorShape::GetNumDimensions().

         : m_Shape(shape)
         , m_Data(data)
         , m_DataLayout(dataLayout)
     {
         ARMNN_ASSERT(m_Shape.GetNumDimensions() == 4);
     }

Member Function Documentation

◆ Get()

DataType& Get	(	unsigned int	b,
		unsigned int	c,
		unsigned int	h,
		unsigned int	w
	)		const

inline

Definition at line 31 of file TensorBufferArrayView.hpp.

References DataLayoutIndexed::GetIndex().

     {
         return m_Data[m_DataLayout.GetIndex(m_Shape, b, c, h, w)];
     }

The documentation for this class was generated from the following file:

src/backends/reference/workloads/TensorBufferArrayView.hpp