21.05
|
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 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 860 of file Descriptors.hpp.
|
inline |
Definition at line 862 of file Descriptors.hpp.
References armnn::NCHW.
|
inline |
Definition at line 868 of file Descriptors.hpp.
|
inline |
Definition at line 875 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 883 of file Descriptors.hpp.
Referenced by ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload(), TfLiteParserImpl::CreateNetworkFromBinary(), SpaceToBatchNdLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), NeonSpaceToBatchNdWorkload::NeonSpaceToBatchNdWorkload(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), and armnn::SpaceToBatchNd().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 888 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), TfLiteParserImpl::CreateNetworkFromBinary(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), and armnn::SpaceToBatchNd().
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 886 of file Descriptors.hpp.
Referenced by TfLiteParserImpl::CreateNetworkFromBinary(), SpaceToBatchNdLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), SpaceToBatchNdLayer::InferOutputShapes(), SpaceToBatchNdDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToBatchNdDescriptor >::Serialize(), and armnn::SpaceToBatchNd().