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.
std::vector< int >	m_End
	End values for the input that will be sliced.
std::vector< int >	m_Stride
	Stride values for the input that will be sliced.
int32_t	m_BeginMask
	Begin mask value.
int32_t	m_EndMask
	End mask value.
int32_t	m_ShrinkAxisMask
	Shrink axis mask value. If set, the nth specification shrinks the dimensionality by 1.
int32_t	m_EllipsisMask
	Ellipsis mask value.
int32_t	m_NewAxisMask
	New axis mask value.
DataLayout	m_DataLayout
	The data layout to be used (NCHW, NHWC).

Detailed Description

A StridedSliceDescriptor for the StridedSliceLayer.

Definition at line 1303 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 1305 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 m_Begin, m_BeginMask, m_DataLayout, m_EllipsisMask, m_End, m_EndMask, m_NewAxisMask, m_ShrinkAxisMask, m_Stride, and armnn::NCHW.

Referenced by operator==(), and StridedSliceDescriptor().

◆ StridedSliceDescriptor() [2/2]

StridedSliceDescriptor ( )

inline

Definition at line 1319 of file Descriptors.hpp.

1320 : StridedSliceDescriptor({}, {}, {})

1321 {}

References StridedSliceDescriptor().

Member Function Documentation

◆ GetStartForAxis()

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

Definition at line 393 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 m_Begin, m_BeginMask, m_Stride, and armnn::numeric_cast().

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

◆ GetStopForAxis()

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

Definition at line 420 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 m_End, m_EndMask, m_ShrinkAxisMask, m_Stride, and armnn::numeric_cast().

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

◆ operator==()

bool operator== ( const StridedSliceDescriptor & rhs ) const

inline

Definition at line 1323 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;
    }