ArmNN
NotReleased
|
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... | |
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
Definition at line 810 of file Descriptors.hpp.
|
inline |
Definition at line 812 of file Descriptors.hpp.
|
inline |
Definition at line 816 of file Descriptors.hpp.
|
inline |
Definition at line 821 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 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().
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().