22.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 1056 of file Descriptors.hpp.
|
inline |
Definition at line 1058 of file Descriptors.hpp.
|
inline |
Definition at line 1062 of file Descriptors.hpp.
|
inline |
Definition at line 1067 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 1073 of file Descriptors.hpp.
Referenced by 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(), SpaceToDepthInferOutputShapeTest(), and TEST_SUITE().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 1076 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(), SpaceToDepthInferOutputShapeTest(), and TEST_SUITE().