23.05
|
A SpaceToDepthDescriptor for the SpaceToDepthLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
SpaceToDepthDescriptor () | |
SpaceToDepthDescriptor (unsigned int blockSize, DataLayout dataLayout) | |
bool | operator== (const SpaceToDepthDescriptor &rhs) const |
Public Member Functions inherited from BaseDescriptor | |
virtual bool | IsNull () const |
virtual | ~BaseDescriptor ()=default |
Public Attributes | |
unsigned int | m_BlockSize |
Scalar specifying the input block size. It must be >= 1. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition at line 1042 of file Descriptors.hpp.
|
inline |
|
inline |
Definition at line 1048 of file Descriptors.hpp.
|
inline |
Definition at line 1053 of file Descriptors.hpp.
References SpaceToDepthDescriptor::m_BlockSize, and SpaceToDepthDescriptor::m_DataLayout.
unsigned int m_BlockSize |
Scalar specifying the input block size. It must be >= 1.
Definition at line 1059 of file Descriptors.hpp.
Referenced by DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), SpaceToDepthDescriptor::operator==(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 1062 of file Descriptors.hpp.
Referenced by armnn::ClDepthToSpaceWorkloadValidate(), armnn::ClSpaceToDepthWorkloadValidate(), SpaceToDepthLayer::CreateWorkload(), DepthToSpaceLayer::CreateWorkload(), armnn::DepthToSpace(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), armnn::NeonDepthToSpaceWorkloadValidate(), armnn::NeonSpaceToDepthWorkloadValidate(), SpaceToDepthDescriptor::operator==(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().