#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()

template<typename DataType>

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 armnn::NCHW.

Member Function Documentation

◆ Get()