21.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 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 892 of file Descriptors.hpp.
|
inline |
Definition at line 894 of file Descriptors.hpp.
|
inline |
Definition at line 898 of file Descriptors.hpp.
|
inline |
Definition at line 903 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 909 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 912 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().