21.02
|
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 884 of file Descriptors.hpp.
|
inline |
Definition at line 886 of file Descriptors.hpp.
|
inline |
Definition at line 890 of file Descriptors.hpp.
|
inline |
Definition at line 895 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 901 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceTest1(), DepthToSpaceTest2(), DepthToSpaceTest3(), DepthToSpaceTest4(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), SpaceToDepthDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), and SpaceToDepthInferOutputShapeTest().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 904 of file Descriptors.hpp.
Referenced by armnn::ClDepthToSpaceWorkloadValidate(), armnn::ClSpaceToDepthWorkloadValidate(), DepthToSpaceLayer::CreateWorkload(), SpaceToDepthLayer::CreateWorkload(), armnn::DepthToSpace(), DepthToSpaceTest1(), DepthToSpaceTest2(), DepthToSpaceTest3(), DepthToSpaceTest4(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), SpaceToDepthLayer::InferOutputShapes(), DepthToSpaceLayer::InferOutputShapes(), armnn::NeonDepthToSpaceWorkloadValidate(), armnn::NeonSpaceToDepthWorkloadValidate(), SpaceToDepthDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SpaceToDepthDescriptor >::Serialize(), armnn::SpaceToDepth(), and SpaceToDepthInferOutputShapeTest().