An OriginsDescriptor for the ConcatLayer. More...

#include <Descriptors.hpp>

Public Member Functions
	OriginsDescriptor ()

	OriginsDescriptor (uint32_t numViews, uint32_t numDimensions=4)

	OriginsDescriptor (const OriginsDescriptor &other)

	OriginsDescriptor (OriginsDescriptor &&other)

	~OriginsDescriptor ()

OriginsDescriptor &	operator= (OriginsDescriptor rhs)

bool	operator== (const OriginsDescriptor &rhs) const

Status	SetViewOriginCoord (uint32_t view, uint32_t coord, uint32_t value)
	Set the view origin coordinates. More...

uint32_t	GetNumViews () const
	Get the number of views. More...

uint32_t	GetNumDimensions () const
	Get the number of dimensions. More...

const uint32_t *	GetViewOrigin (uint32_t idx) const
	Return the view origin at the int value idx. More...

void	ReorderOrigins (unsigned int *newOrdering, unsigned int numNewOrdering)
	Reorders the viewOrigins in accordance with the indices presented in newOrdering array. More...

void	SetConcatAxis (unsigned int concatAxis)
	Set the concatenation axis value. More...

unsigned int	GetConcatAxis () const
	Get the concatenation axis value. More...

Friends
void	swap (OriginsDescriptor &first, OriginsDescriptor &second)
	Swap the ViewsDescriptor values first and second. More...

Detailed Description

An OriginsDescriptor for the ConcatLayer.

Descriptor to configure the concatenation process. Number of views must be equal to the number of inputs, and their order must match - e.g. first view corresponds to the first input, second view to the second input, etc.

Definition at line 155 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ OriginsDescriptor() [1/4]

OriginsDescriptor ( )

Definition at line 76 of file Descriptors.cpp.

 : m_ConcatAxis(1)
 , m_NumViews(0)
 , m_NumDimensions(0)
 , m_ViewOrigins(nullptr)
 {}

◆ OriginsDescriptor() [2/4]

OriginsDescriptor	(	uint32_t	numViews,
		uint32_t	numDimensions = `4`
	)

Definition at line 83 of file Descriptors.cpp.

 : m_ConcatAxis(1)
 , m_NumViews(numViews)
 , m_NumDimensions(numDimensions)
 , m_ViewOrigins(numViews && numDimensions > 0 ? new uint32_t *[numViews]() : nullptr)
 {
     for (uint32_t i = 0; m_NumDimensions > 0 && i < m_NumViews; ++i)
     {
         m_ViewOrigins[i] = new uint32_t[m_NumDimensions]();
     }
 }

◆ OriginsDescriptor() [3/4]

OriginsDescriptor ( const OriginsDescriptor & other )

Definition at line 95 of file Descriptors.cpp.

 : m_ConcatAxis(other.m_ConcatAxis)
 , m_NumViews(other.m_NumViews)
 , m_NumDimensions(other.m_NumDimensions)
 , m_ViewOrigins(other.m_NumViews && other.m_NumDimensions > 0 ? new uint32_t *[other.m_NumViews]() : nullptr)
 {
     for (uint32_t i = 0; m_NumDimensions > 0 && i < m_NumViews; ++i)
     {
         m_ViewOrigins[i] = new uint32_t[m_NumDimensions]();
         memcpy(m_ViewOrigins[i], other.m_ViewOrigins[i], m_NumDimensions * sizeof(uint32_t));
     }
 }

◆ OriginsDescriptor() [4/4]

OriginsDescriptor ( OriginsDescriptor && other )

Definition at line 108 of file Descriptors.cpp.

References OriginsDescriptor::swap.

 : OriginsDescriptor()
 {
     swap(*this, other);
 }

◆ ~OriginsDescriptor()

~OriginsDescriptor ( )

Definition at line 114 of file Descriptors.cpp.

 {
     for (uint32_t i = 0; m_NumDimensions > 0 && i < m_NumViews; ++i)
     {
         delete[] m_ViewOrigins[i];
     }
     delete[] m_ViewOrigins;
 }

Member Function Documentation

◆ GetConcatAxis()

unsigned int GetConcatAxis ( ) const

Get the concatenation axis value.

Definition at line 156 of file Descriptors.cpp.

Referenced by ClLayerSupport::IsConcatSupported(), NeonLayerSupport::IsConcatSupported(), OriginsDescriptor::operator==(), StringifyLayerParameters< OriginsDescriptor >::Serialize(), SerializerVisitor::VisitConcatLayer(), and SerializerVisitor::VisitSplitterLayer().

 {
     return m_ConcatAxis;
 }

◆ GetNumDimensions()

uint32_t GetNumDimensions ( ) const

Get the number of dimensions.

Definition at line 186 of file Descriptors.cpp.

Referenced by Concatenate(), ConcatLayer::CreateWorkload(), ViewsDescriptor::GetNumDimensions(), ConcatLayer::InferOutputShapes(), NeonLayerSupport::IsConcatSupported(), ClLayerSupport::IsConcatSupported(), OriginsDescriptor::operator==(), armnnUtils::ProcessConcatInputTensorInfo(), StringifyLayerParameters< OriginsDescriptor >::Serialize(), SerializerVisitor::VisitConcatLayer(), and SerializerVisitor::VisitSplitterLayer().

 {
     return m_NumDimensions;
 }

◆ GetNumViews()

uint32_t GetNumViews ( ) const

Get the number of views.

Definition at line 181 of file Descriptors.cpp.

Referenced by Concatenate(), ConcatLayer::CreateWorkload(), ViewsDescriptor::GetNumViews(), ConcatLayer::InferOutputShapes(), OriginsDescriptor::operator==(), StringifyLayerParameters< OriginsDescriptor >::Serialize(), ConcatLayer::ValidateTensorShapesFromInputs(), SerializerVisitor::VisitConcatLayer(), and SerializerVisitor::VisitSplitterLayer().

 {
     return m_NumViews;
 }

◆ GetViewOrigin()

const uint32_t * GetViewOrigin ( uint32_t idx ) const

Return the view origin at the int value idx.

Definition at line 191 of file Descriptors.cpp.

Referenced by armnnTfParser::CalculatePaddedOutputTensorInfo(), Concatenate(), ConcatLayer::CreateWorkload(), ViewsDescriptor::GetViewOrigin(), ConcatLayer::InferOutputShapes(), OriginsDescriptor::operator==(), StringifyLayerParameters< OriginsDescriptor >::Serialize(), and SerializerVisitor::VisitConcatLayer().

 {
     return m_ViewOrigins ? m_ViewOrigins[idx] : nullptr;
 }

◆ operator=()

OriginsDescriptor & operator= ( OriginsDescriptor rhs )

Definition at line 123 of file Descriptors.cpp.

References OriginsDescriptor::swap.

 {
     swap(*this, rhs);
     return *this;
 }

◆ operator==()

bool operator== ( const OriginsDescriptor & rhs ) const

Definition at line 129 of file Descriptors.cpp.

References OriginsDescriptor::GetConcatAxis(), OriginsDescriptor::GetNumDimensions(), OriginsDescriptor::GetNumViews(), and OriginsDescriptor::GetViewOrigin().

 {
     if (GetNumViews()      != rhs.GetNumViews() ||
         GetNumDimensions() != rhs.GetNumDimensions() ||
         GetConcatAxis()    != rhs.GetConcatAxis())
     {
         return false;
     }
 
     for (unsigned int i = 0u; i < GetNumViews(); ++i)
     {
         for (unsigned int j = 0u; j < GetNumDimensions(); ++j)
         {
             if (GetViewOrigin(i)[j] != rhs.GetViewOrigin(i)[j])
             {
                 return false;
             }
         }
     }
 
     return true;
 }

◆ ReorderOrigins()

void ReorderOrigins	(	unsigned int *	newOrdering,
		unsigned int	numNewOrdering
	)

Reorders the viewOrigins in accordance with the indices presented in newOrdering array.

The number of views must match number of elements in the new ordering array.

Definition at line 198 of file Descriptors.cpp.

References ARMNN_ASSERT_MSG.

 {
     ARMNN_ASSERT_MSG(m_NumViews == numNewOrdering, "number of views must match number of "
         "elements in the new ordering array");
     std::vector<uint32_t*> viewOrigins(&m_ViewOrigins[0], &m_ViewOrigins[m_NumViews]);
 
     for (unsigned int i = 0; i < numNewOrdering; ++i)
     {
         m_ViewOrigins[i] = viewOrigins[newOrdering[i]];
     }
 }

◆ SetConcatAxis()

void SetConcatAxis ( unsigned int concatAxis )

Set the concatenation axis value.

Definition at line 152 of file Descriptors.cpp.

Referenced by armnnTfParser::CalculatePaddedOutputTensorInfo(), armnn::CreateDescriptorForConcatenation(), and TfLiteParser::OutputShapeOfReshape().

 {
     m_ConcatAxis = concatAxis;
 }

◆ SetViewOriginCoord()

Status SetViewOriginCoord	(	uint32_t	view,
		uint32_t	coord,
		uint32_t	value
	)

Set the view origin coordinates.

The arguments are: view, dimension, value. If the view is greater than or equal to GetNumViews(), then the view argument is out of range. If the coord is greater than or equal to GetNumDimensions(), then the coord argument is out of range.

Definition at line 161 of file Descriptors.cpp.

References ARMNN_LOG, armnn::error, armnn::Failure, and armnn::Success.

Referenced by CaffeParserBase::AddConvLayerWithSplits(), armnnTfParser::CalculatePaddedOutputTensorInfo(), armnn::CreateDescriptorForConcatenation(), CaffeParserBase::ParseConcatLayer(), armnnUtils::ProcessConcatInputTensorInfo(), and ViewsDescriptor::SetViewOriginCoord().

 {
     if (view >= m_NumViews)
     {
         ARMNN_LOG(error) << "OriginsDescriptor::SetViewOriginCoord: view argument:" << view <<
             " is out of range";
         return Status::Failure;
     }
     if (coord >= m_NumDimensions)
     {
         ARMNN_LOG(error) << "OriginsDescriptor::SetViewOriginCoord: coord argument:" << coord <<
             " is out of range";
         return Status::Failure;
     }
 
     m_ViewOrigins[view][coord] = value;
     return Status::Success;
 }

Friends And Related Function Documentation

◆ swap

void swap	(	OriginsDescriptor &	first,
		OriginsDescriptor &	second
	)

friend

Swap the ViewsDescriptor values first and second.

Definition at line 344 of file Descriptors.cpp.

Referenced by OriginsDescriptor::operator=(), and OriginsDescriptor::OriginsDescriptor().

 {
     using std::swap;
     swap(first.m_NumViews, second.m_NumViews);
     swap(first.m_NumDimensions, second.m_NumDimensions);
     swap(first.m_ViewOrigins, second.m_ViewOrigins);
     swap(first.m_ConcatAxis, second.m_ConcatAxis);
 }

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

include/armnn/Descriptors.hpp
src/armnn/Descriptors.cpp

Public Member Functions

Friends