23.02
|
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
BatchToSpaceNdDescriptor () | |
BatchToSpaceNdDescriptor (std::vector< unsigned int > blockShape, std::vector< std::pair< unsigned int, unsigned int >> crops) | |
bool | operator== (const BatchToSpaceNdDescriptor &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 values. More... | |
std::vector< std::pair< unsigned int, unsigned int > > | m_Crops |
The values to crop from the input dimension. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition at line 843 of file Descriptors.hpp.
|
inline |
Definition at line 845 of file Descriptors.hpp.
|
inline |
Definition at line 851 of file Descriptors.hpp.
References armnn::NCHW.
|
inline |
Definition at line 858 of file Descriptors.hpp.
References BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, and BatchToSpaceNdDescriptor::m_DataLayout.
std::vector<unsigned int> m_BlockShape |
Block shape values.
Definition at line 866 of file Descriptors.hpp.
Referenced by BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), and StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize().
std::vector<std::pair<unsigned int, unsigned int> > m_Crops |
The values to crop from the input dimension.
Definition at line 868 of file Descriptors.hpp.
Referenced by BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), and StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 870 of file Descriptors.hpp.
Referenced by armnn::ClBatchToSpaceNdWorkloadValidate(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), and StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize().