SpaceToDepthDescriptor Struct Reference

A SpaceToDepthDescriptor for the SpaceToDepthLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for SpaceToDepthDescriptor:

Collaboration diagram for SpaceToDepthDescriptor:

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...

Detailed Description

A SpaceToDepthDescriptor for the SpaceToDepthLayer.

Definition at line 1075 of file Descriptors.hpp.

Constructor & Destructor Documentation

SpaceToDepthDescriptor() [1/2]

SpaceToDepthDescriptor ( )

inline

Definition at line 1077 of file Descriptors.hpp.

        : SpaceToDepthDescriptor(1u, DataLayout::NHWC)
    {}

References armnn::NHWC.

SpaceToDepthDescriptor() [2/2]

SpaceToDepthDescriptor ( unsigned int  blockSize, DataLayout  dataLayout  )

inline

Definition at line 1081 of file Descriptors.hpp.

        : m_BlockSize(blockSize)
        , m_DataLayout(dataLayout)
    {}

Member Function Documentation

operator==()

bool operator== ( const SpaceToDepthDescriptor &  rhs ) const

inline

Definition at line 1086 of file Descriptors.hpp.

    {
        return m_BlockSize == rhs.m_BlockSize && m_DataLayout == rhs.m_DataLayout;
    }

References SpaceToDepthDescriptor::m_BlockSize, and SpaceToDepthDescriptor::m_DataLayout.

Member Data Documentation

m_BlockSize

unsigned int m_BlockSize

Scalar specifying the input block size. It must be >= 1.

Definition at line 1092 of file Descriptors.hpp.

Referenced by DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceLayer::InferOutputShapes(), SpaceToDepthLayer::InferOutputShapes(), SpaceToDepthDescriptor::operator==(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().

m_DataLayout

DataLayout m_DataLayout

The data layout to be used (NCHW, NHWC).

Definition at line 1095 of file Descriptors.hpp.

Referenced by armnn::ClDepthToSpaceWorkloadValidate(), armnn::ClSpaceToDepthWorkloadValidate(), ConvertDepthToSpaceToTosaOperator(), DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceLayer::InferOutputShapes(), SpaceToDepthLayer::InferOutputShapes(), armnn::NeonDepthToSpaceWorkloadValidate(), armnn::NeonSpaceToDepthWorkloadValidate(), SpaceToDepthDescriptor::operator==(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthQueueDescriptor::Validate(), and DepthToSpaceQueueDescriptor::Validate().

---

The documentation for this struct was generated from the following file:

• include/armnn/Descriptors.hpp