22.08
|
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
SpaceToBatchNdDescriptor () | |
SpaceToBatchNdDescriptor (const std::vector< unsigned int > &blockShape, const std::vector< std::pair< unsigned int, unsigned int >> &padList) | |
bool | operator== (const SpaceToBatchNdDescriptor &rhs) const |
Public Member Functions inherited from BaseDescriptor | |
virtual bool | IsNull () const |
virtual | ~BaseDescriptor ()=default |
Public Attributes | |
std::vector< unsigned int > | m_BlockShape |
Block shape value. More... | |
std::vector< std::pair< unsigned int, unsigned int > > | m_PadList |
Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left, right}. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
Definition at line 1020 of file Descriptors.hpp.
|
inline |
Definition at line 1022 of file Descriptors.hpp.
References armnn::NCHW.
|
inline |
Definition at line 1028 of file Descriptors.hpp.
|
inline |
Definition at line 1035 of file Descriptors.hpp.
References SpaceToBatchNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_DataLayout, and SpaceToBatchNdDescriptor::m_PadList.
std::vector<unsigned int> m_BlockShape |
Block shape value.
Definition at line 1043 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::AsFloatArray(), ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload(), SpaceToBatchNdLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), NeonSpaceToBatchNdWorkload::NeonSpaceToBatchNdWorkload(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and TEST_SUITE().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 1048 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::AsFloatArray(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and TEST_SUITE().
std::vector<std::pair<unsigned int, unsigned int> > m_PadList |
Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left, right}.
Definition at line 1046 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::AsFloatArray(), SpaceToBatchNdLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), armnn::SpaceToBatchNd(), and TEST_SUITE().