ArmNN
 20.02
Public Member Functions | Public Attributes | List of all members
SpaceToDepthDescriptor Struct Reference

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 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 810 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ SpaceToDepthDescriptor() [1/2]

SpaceToDepthDescriptor ( )
inline

Definition at line 812 of file Descriptors.hpp.

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

◆ SpaceToDepthDescriptor() [2/2]

SpaceToDepthDescriptor ( unsigned int  blockSize,
DataLayout  dataLayout 
)
inline

Definition at line 816 of file Descriptors.hpp.

817  : m_BlockSize(blockSize)
818  , m_DataLayout(dataLayout)
819  {}
armnn::SpaceToDepthDescriptor::m_BlockSize
unsigned int m_BlockSize
Scalar specifying the input block size. It must be >= 1.
Definition: Descriptors.hpp:827
armnn::SpaceToDepthDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:830

Member Function Documentation

◆ operator==()

bool operator== ( const SpaceToDepthDescriptor &  rhs) const
inline

Definition at line 821 of file Descriptors.hpp.

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

822  {
823  return m_BlockSize == rhs.m_BlockSize && m_DataLayout == rhs.m_DataLayout;
824  }
armnn::SpaceToDepthDescriptor::m_BlockSize
unsigned int m_BlockSize
Scalar specifying the input block size. It must be >= 1.
Definition: Descriptors.hpp:827
armnn::SpaceToDepthDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:830

Member Data Documentation

◆ m_BlockSize

unsigned int m_BlockSize

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

Definition at line 827 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), SpaceToDepthLayer::CreateWorkload(), DepthToSpaceLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceTest1(), DepthToSpaceTest2(), DepthToSpaceTest3(), DepthToSpaceTest4(), Deserializer::GetNetworkOutputBindingInfo(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), SpaceToDepthDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthInferOutputShapeTest(), SerializerVisitor::VisitDepthToSpaceLayer(), and SerializerVisitor::VisitSpaceToDepthLayer().

◆ m_DataLayout

DataLayout m_DataLayout

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

Definition at line 830 of file Descriptors.hpp.

Referenced by armnn::ClDepthToSpaceWorkloadValidate(), armnn::ClSpaceToDepthWorkloadValidate(), DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceTest1(), DepthToSpaceTest2(), DepthToSpaceTest3(), DepthToSpaceTest4(), Deserializer::GetNetworkOutputBindingInfo(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), armnn::NeonDepthToSpaceWorkloadValidate(), armnn::NeonSpaceToDepthWorkloadValidate(), SpaceToDepthDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), SpaceToDepthInferOutputShapeTest(), SerializerVisitor::VisitDepthToSpaceLayer(), and SerializerVisitor::VisitSpaceToDepthLayer().

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