+ +

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 1014 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 1016 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 1030 of file Descriptors.hpp.

+

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

1032 {}

armnn::StridedSliceDescriptor::StridedSliceDescriptor

StridedSliceDescriptor()

Definition: Descriptors.hpp:1030

+

Member Function Documentation

+ +

◆ GetStartForAxis()

+ +

+

+ + + + + + + + + + + + + + + + + + +

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

+

+ +

Definition at line 360 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 387 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 1034 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 1053 of file Descriptors.hpp.

+ +

Referenced by ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), StridedSliceLayer::InferOutputShapes(), 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 1061 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 1074 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 1068 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 1055 of file Descriptors.hpp.

+ +

Referenced by ClStridedSliceWorkload::ClStridedSliceWorkload(), StridedSliceLayer::CreateWorkload(), StridedSliceLayer::InferOutputShapes(), 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 1064 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 1071 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 1066 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 1057 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

+

+Public Member Functions

+Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ StridedSliceDescriptor() [1/2]

◆ StridedSliceDescriptor() [2/2]

Member Function Documentation

◆ GetStartForAxis()

◆ GetStopForAxis()

◆ operator==()

Member Data Documentation

◆ m_Begin

◆ m_BeginMask

◆ m_DataLayout

◆ m_EllipsisMask

◆ m_End

◆ m_EndMask

◆ m_NewAxisMask

◆ m_ShrinkAxisMask

◆ m_Stride