BaseTensor< MemoryType > Class Template Reference

#include <Tensor.hpp>

Public Member Functions
	BaseTensor ()
	Empty (invalid) constructor. More...
	BaseTensor (const TensorInfo &info, MemoryType memoryArea)
	Constructor from a raw memory pointer. More...
	BaseTensor (const BaseTensor &other)
	Tensors are copyable. More...
BaseTensor &	operator= (const BaseTensor &)
	Tensors are copyable. More...
const TensorInfo &	GetInfo () const
TensorInfo &	GetInfo ()
const TensorShape &	GetShape () const
TensorShape &	GetShape ()
DataType	GetDataType () const
unsigned int	GetNumDimensions () const
unsigned int	GetNumBytes () const
unsigned int	GetNumElements () const
MemoryType	GetMemoryArea () const

Protected Member Functions
	~BaseTensor ()
	Protected destructor to stop users from making these (could still new one on the heap and then leak it...) More...

Protected Attributes
MemoryType	m_MemoryArea

Detailed Description

template<typename MemoryType>
class armnn::BaseTensor< MemoryType >

Definition at line 264 of file Tensor.hpp.

Constructor & Destructor Documentation

BaseTensor() [1/3]

BaseTensor

(

)

Empty (invalid) constructor.

Definition at line 505 of file Tensor.cpp.

 : m_MemoryArea(nullptr)
{
}

BaseTensor() [2/3]

BaseTensor	(	const TensorInfo &	info,
		MemoryType	memoryArea
	)

Constructor from a raw memory pointer.

Parameters

memoryArea

- Region of CPU-addressable memory where tensor data will be stored. Must be valid while workloads are on the fly. Tensor instances do not claim ownership of referenced memory regions, that is, no attempt will be made by ArmNN to free these memory regions automatically.

Definition at line 511 of file Tensor.cpp.

 : m_MemoryArea(memoryArea)
 , m_Info(info)
{
}

BaseTensor() [3/3]

BaseTensor

(

const BaseTensor< MemoryType > &

other

)

Tensors are copyable.

~BaseTensor()

~BaseTensor ( )

inlineprotected

Protected destructor to stop users from making these (could still new one on the heap and then leak it...)

Definition at line 297 of file Tensor.hpp.

{}

Member Function Documentation

GetDataType()

DataType GetDataType ( ) const

inline

Definition at line 287 of file Tensor.hpp.

Referenced by CompareConstTensor(), and armnn::ConvertWeightTensorFromArmnnToAcl().

{ return m_Info.GetDataType(); }

GetInfo() [1/2]

const TensorInfo& GetInfo ( ) const

inline

Definition at line 282 of file Tensor.hpp.

Referenced by BaseTensor< const void *>::BaseTensor(), BOOST_AUTO_TEST_CASE(), TestLayerVisitor::CheckConstTensors(), CheckTensor(), armnn::ConvertWeightTensorFromArmnnToAcl(), armnn::CreateQuantizedConst(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::GetInputTensorInfo(), QuantizerStrategy::QuantizerStrategy(), armnn::ReorderWeightChannelsForAcl(), and LayerVerifierBase::VerifyConstTensors().

{ return m_Info; }

GetInfo() [2/2]

TensorInfo& GetInfo ( )

inline

Definition at line 283 of file Tensor.hpp.

{ return m_Info; }

GetMemoryArea()

MemoryType GetMemoryArea ( ) const

inline

Definition at line 292 of file Tensor.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), TestLayerVisitor::CheckConstTensors(), CompareConstTensor(), armnn::CreateQuantizedConst(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), ScopedCpuTensorHandle::ScopedCpuTensorHandle(), and LayerVerifierBase::VerifyConstTensors().

{ return m_MemoryArea; }

GetNumBytes()

unsigned int GetNumBytes ( ) const

inline

Definition at line 289 of file Tensor.hpp.

Referenced by TestLayerVisitor::CheckConstTensors(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), ScopedCpuTensorHandle::ScopedCpuTensorHandle(), and LayerVerifierBase::VerifyConstTensors().

{ return m_Info.GetNumBytes(); }

GetNumDimensions()

unsigned int GetNumDimensions ( ) const

inline

Definition at line 288 of file Tensor.hpp.

Referenced by TestLayerVisitor::CheckConstTensors().

{ return m_Info.GetNumDimensions(); }

GetNumElements()

unsigned int GetNumElements ( ) const

inline

Definition at line 290 of file Tensor.hpp.

Referenced by TestLayerVisitor::CheckConstTensors(), and CompareConstTensor().

{ return m_Info.GetNumElements(); }

GetShape() [1/2]

const TensorShape& GetShape ( ) const

inline

Definition at line 284 of file Tensor.hpp.

Referenced by ITfParser::TfParserImpl::AddFullyConnectedLayer(), BOOST_AUTO_TEST_CASE(), armnnTfParser::CheckPaddingTensor(), CompareConstTensor(), ITfParser::TfParserImpl::ParseConv2D(), ITfParser::TfParserImpl::ParseDepthwiseConv2D(), and armnn::ReorderWeightChannelsForAcl().

{ return m_Info.GetShape(); }

GetShape() [2/2]

TensorShape& GetShape ( )

inline

Definition at line 285 of file Tensor.hpp.

{ return m_Info.GetShape(); }

operator=()

BaseTensor< MemoryType > & operator=

(

const BaseTensor< MemoryType > &

)

Tensors are copyable.

Definition at line 525 of file Tensor.cpp.

{
    m_Info = other.m_Info;
    m_MemoryArea = other.m_MemoryArea;
    return *this;
}

Member Data Documentation

m_MemoryArea

MemoryType m_MemoryArea

protected

Definition at line 299 of file Tensor.hpp.

Referenced by BaseTensor< const void *>::BaseTensor(), and BaseTensor< const void *>::operator=().

---

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

• include/armnn/Tensor.hpp
• src/armnn/Tensor.cpp