ArmNN
NotReleased
|
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
BatchToSpaceNdDescriptor () | |
BatchToSpaceNdDescriptor (std::vector< unsigned int > blockShape, std::vector< std::pair< unsigned int, unsigned int >> crops) | |
bool | operator== (const BatchToSpaceNdDescriptor &rhs) const |
Public Attributes | |
std::vector< unsigned int > | m_BlockShape |
Block shape values. More... | |
std::vector< std::pair< unsigned int, unsigned int > > | m_Crops |
The values to crop from the input dimension. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition at line 657 of file Descriptors.hpp.
|
inline |
Definition at line 659 of file Descriptors.hpp.
References armnn::NCHW.
|
inline |
Definition at line 665 of file Descriptors.hpp.
|
inline |
Definition at line 672 of file Descriptors.hpp.
References BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, and BatchToSpaceNdDescriptor::m_DataLayout.
std::vector<unsigned int> m_BlockShape |
Block shape values.
Definition at line 680 of file Descriptors.hpp.
Referenced by BatchToSpaceInferOutputShapeTest(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().
std::vector<std::pair<unsigned int, unsigned int> > m_Crops |
The values to crop from the input dimension.
Definition at line 682 of file Descriptors.hpp.
Referenced by BatchToSpaceInferOutputShapeTest(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 684 of file Descriptors.hpp.
Referenced by BatchToSpaceInferOutputShapeTest(), BOOST_AUTO_TEST_CASE(), armnn::ClBatchToSpaceNdWorkloadValidate(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().