StridedSliceDescriptor Struct Reference

A StridedSliceDescriptor for the StridedSliceLayer. More...

#include <Descriptors.hpp>

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 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 1002 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 1004 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)
    {}

StridedSliceDescriptor() [2/2]

StridedSliceDescriptor ( )

inline

Definition at line 1018 of file Descriptors.hpp.

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

Member Function Documentation

GetStartForAxis()

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

Definition at line 358 of file Descriptors.cpp.

References armnn::numeric_cast().

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

{
    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 = boost::numeric_cast<int>(inputShape[axis]);
    if (start < 0)
    {
        start += (axisSize);
    }

    return std::max(0, std::min(start, axisSize - 1));

}

GetStopForAxis()

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

Definition at line 385 of file Descriptors.cpp.

References armnn::numeric_cast().

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

{

    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 = boost::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));

}

operator==()

bool operator== ( const StridedSliceDescriptor &  rhs ) const

inline

Definition at line 1022 of file Descriptors.hpp.

References StridedSliceDescriptor::m_Begin, StridedSliceDescriptor::m_BeginMask, StridedSliceDescriptor::m_DataLayout, StridedSliceDescriptor::m_EllipsisMask, StridedSliceDescriptor::m_End, StridedSliceDescriptor::m_EndMask, StridedSliceDescriptor::m_NewAxisMask, StridedSliceDescriptor::m_ShrinkAxisMask, and StridedSliceDescriptor::m_Stride.

    {
        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;
    }

Member Data Documentation

m_Begin

std::vector<int> m_Begin

Begin values for the input that will be sliced.

Definition at line 1041 of file Descriptors.hpp.

Referenced by ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_BeginMask

int32_t m_BeginMask

Begin mask value.

If set, then the begin is disregarded and the fullest range is used for the dimension.

Definition at line 1049 of file Descriptors.hpp.

Referenced by ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_DataLayout

DataLayout m_DataLayout

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

Definition at line 1062 of file Descriptors.hpp.

Referenced by StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_EllipsisMask

int32_t m_EllipsisMask

Ellipsis mask value.

Definition at line 1056 of file Descriptors.hpp.

Referenced by StridedSliceLayer::CreateWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_End

std::vector<int> m_End

End values for the input that will be sliced.

Definition at line 1043 of file Descriptors.hpp.

Referenced by ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_EndMask

int32_t m_EndMask

End mask value.

If set, then the end is disregarded and the fullest range is used for the dimension.

Definition at line 1052 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_NewAxisMask

int32_t m_NewAxisMask

New axis mask value.

If set, the begin, end and stride is disregarded and a new 1 dimension is inserted to this location of the output tensor.

Definition at line 1059 of file Descriptors.hpp.

Referenced by StridedSliceLayer::CreateWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_ShrinkAxisMask

int32_t m_ShrinkAxisMask

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

Definition at line 1054 of file Descriptors.hpp.

Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), StridedSliceLayer::InferOutputShapes(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

m_Stride

std::vector<int> m_Stride

Stride values for the input that will be sliced.

Definition at line 1045 of file Descriptors.hpp.

Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), StridedSliceLayer::InferOutputShapes(), NeonStridedSliceWorkload::NeonStridedSliceWorkload(), StridedSliceDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< StridedSliceDescriptor >::Serialize(), and SerializerVisitor::VisitStridedSliceLayer().

---

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

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