TensorBufferArrayView< DataType > Class Template Reference

#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 TensorShape::GetNumDimensions().

Member Function Documentation

Get()

template<typename DataType >

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