ArmNN
NotReleased
|
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. If set, then the begin is disregarded and the fullest range is used for the dimension. More... | |
int32_t | m_EndMask |
End mask value. If set, then the end is disregarded and the fullest range is used for the dimension. 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. If set, the begin, end and stride is disregarded and a new 1 dimension is inserted to this location of the output tensor. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A StridedSliceDescriptor for the StridedSliceLayer.
Definition at line 1002 of file Descriptors.hpp.
|
inline |
Definition at line 1004 of file Descriptors.hpp.
|
inline |
Definition at line 1018 of file Descriptors.hpp.
int GetStartForAxis | ( | const TensorShape & | inputShape, |
unsigned int | axis | ||
) | const |
Definition at line 358 of file Descriptors.cpp.
Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), and StridedSliceLayer::InferOutputShapes().
int GetStopForAxis | ( | const TensorShape & | inputShape, |
unsigned int | axis, | ||
int | startForAxis | ||
) | const |
Definition at line 385 of file Descriptors.cpp.
Referenced by armnnUtils::CalculateStridedSliceOutputTensorInfo(), and StridedSliceLayer::InferOutputShapes().
|
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.
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().
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().
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().
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().
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().
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().
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().
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().
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().