#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 17 of file TensorBufferArrayView.hpp.

Constructor & Destructor Documentation

◆ TensorBufferArrayView()

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

inline

Definition at line 20 of file TensorBufferArrayView.hpp.

         : m_Shape(shape)
         , m_Data(data)
         , m_DataLayout(dataLayout)
     {
         ARMNN_THROW_INVALIDARG_MSG_IF_FALSE(m_Shape.GetNumDimensions() == 4,
                                             "Only $d tensors are supported by TensorBufferArrayView.");
     }

References ARMNN_THROW_INVALIDARG_MSG_IF_FALSE, and TensorShape::GetNumDimensions().

Member Function Documentation

◆ Get()

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

inline

Definition at line 30 of file TensorBufferArrayView.hpp.

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

References DataLayoutIndexed::GetIndex().

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

src/backends/reference/workloads/TensorBufferArrayView.hpp