A StridedSliceDescriptor for the StridedSliceLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for StridedSliceDescriptor:

Collaboration diagram for StridedSliceDescriptor:

Public Member Functions
	StridedSliceDescriptor (const std::vector< int > &begin, const std::vector< int > &end, const std::vector< int > &stride)

	StridedSliceDescriptor ()

bool	operator== (const StridedSliceDescriptor &rhs) const

int	GetStartForAxis (const TensorShape &inputShape, unsigned int axis) const

int	GetStopForAxis (const TensorShape &inputShape, unsigned int axis, int startForAxis) const

Public Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const

virtual	~BaseDescriptor ()=default

Public Attributes
std::vector< int >	m_Begin
	Begin values for the input that will be sliced. More...

std::vector< int >	m_End
	End values for the input that will be sliced. More...

std::vector< int >	m_Stride
	Stride values for the input that will be sliced. More...

int32_t	m_BeginMask
	Begin mask value. More...

int32_t	m_EndMask
	End mask value. More...

int32_t	m_ShrinkAxisMask
	Shrink axis mask value. If set, the nth specification shrinks the dimensionality by 1. More...

int32_t	m_EllipsisMask
	Ellipsis mask value. More...

int32_t	m_NewAxisMask
	New axis mask value. More...

DataLayout	m_DataLayout
	The data layout to be used (NCHW, NHWC). More...

Detailed Description

A StridedSliceDescriptor for the StridedSliceLayer.

Definition at line 1282 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ StridedSliceDescriptor() [1/2]

StridedSliceDescriptor	(	const std::vector< int > &	begin,
		const std::vector< int > &	end,
		const std::vector< int > &	stride
	)

inline

Definition at line 1284 of file Descriptors.hpp.

         : m_Begin(begin)
         , m_End(end)
         , m_Stride(stride)
         , m_BeginMask(0)
         , m_EndMask(0)
         , m_ShrinkAxisMask(0)
         , m_EllipsisMask(0)
         , m_NewAxisMask(0)
         , m_DataLayout(DataLayout::NCHW)
     {}

References armnn::NCHW.

◆ StridedSliceDescriptor() [2/2]

StridedSliceDescriptor ( )

inline

Definition at line 1298 of file Descriptors.hpp.

1299 : StridedSliceDescriptor({}, {}, {})

1300 {}

Member Function Documentation

◆ GetStartForAxis()

int GetStartForAxis	(	const TensorShape &	inputShape,
		unsigned int	axis
	)		const

Definition at line 384 of file Descriptors.cpp.

 {
     int start = m_Begin[axis];
  
     if (m_BeginMask & (1 << axis))
     {
         if (m_Stride[axis] > 0)
         {
             start = std::numeric_limits<int>::min();
         }
         else
         {
             start = std::numeric_limits<int>::max();
         }
     }
  
     const int axisSize = armnn::numeric_cast<int>(inputShape[axis]);
     if (start < 0)
     {
         start += (axisSize);
     }
  
     return std::max(0, std::min(start, axisSize - 1));
  
 }

References StridedSliceDescriptor::m_Begin, StridedSliceDescriptor::m_BeginMask, and StridedSliceDescriptor::m_Stride.

Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), StridedSliceLayer::InferOutputShapes(), and armnn::StridedSlice().

◆ GetStopForAxis()

int GetStopForAxis	(	const TensorShape &	inputShape,
		unsigned int	axis,
		int	startForAxis
	)		const

Definition at line 411 of file Descriptors.cpp.

 {
  
     if (m_ShrinkAxisMask & (1 << axis))
     {
         return startForAxis + 1;
     }
  
     int stop = m_End[axis];
  
     if (m_EndMask & (1 << axis))
     {
         if (m_Stride[axis] > 0)
         {
             stop = std::numeric_limits<int>::max();
         }
         else
         {
             stop = std::numeric_limits<int>::min();
         }
     }
  
     const int axisSize = armnn::numeric_cast<int>(inputShape[axis]);
     if (stop < 0)
     {
         stop += axisSize;
     }
  
     return m_Stride[axis] > 0 ? std::max(0, std::min(stop, axisSize)) :
                                 std::max(-1, std::min(stop, axisSize - 1));
  
 }

References StridedSliceDescriptor::m_End, StridedSliceDescriptor::m_EndMask, StridedSliceDescriptor::m_ShrinkAxisMask, and StridedSliceDescriptor::m_Stride.

Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), StridedSliceLayer::InferOutputShapes(), and armnn::StridedSlice().

◆ operator==()

bool operator== ( const StridedSliceDescriptor & rhs ) const

inline

Definition at line 1302 of file Descriptors.hpp.

     {
         return m_Begin          == rhs.m_Begin &&
                m_End            == rhs.m_End &&
                m_Stride         == rhs.m_Stride &&
                m_BeginMask      == rhs.m_BeginMask &&
                m_EndMask        == rhs.m_EndMask &&
                m_ShrinkAxisMask == rhs.m_ShrinkAxisMask &&
                m_EllipsisMask   == rhs.m_EllipsisMask &&
                m_NewAxisMask    == rhs.m_NewAxisMask &&
                m_DataLayout     == rhs.m_DataLayout;
     }