BatchToSpaceNdDescriptor Struct Reference

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...

Detailed Description

A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.

Definition at line 668 of file Descriptors.hpp.

Constructor & Destructor Documentation

BatchToSpaceNdDescriptor() [1/2]

BatchToSpaceNdDescriptor ( )

inline

Definition at line 670 of file Descriptors.hpp.

References armnn::NCHW.

        : m_BlockShape({1, 1})
        , m_Crops({{0, 0}, {0, 0}})
        , m_DataLayout(DataLayout::NCHW)
    {}

BatchToSpaceNdDescriptor() [2/2]

BatchToSpaceNdDescriptor ( std::vector< unsigned int >  blockShape, std::vector< std::pair< unsigned int, unsigned int >>  crops  )

inline

Definition at line 676 of file Descriptors.hpp.

        : m_BlockShape(blockShape)
        , m_Crops(crops)
        , m_DataLayout(DataLayout::NCHW)
    {}

Member Function Documentation

operator==()

bool operator== ( const BatchToSpaceNdDescriptor &  rhs ) const

inline

Definition at line 683 of file Descriptors.hpp.

References BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, and BatchToSpaceNdDescriptor::m_DataLayout.

    {
        return m_BlockShape == rhs.m_BlockShape &&
               m_Crops      == rhs.m_Crops &&
               m_DataLayout == rhs.m_DataLayout;
    }

Member Data Documentation

m_BlockShape

std::vector<unsigned int> m_BlockShape

Block shape values.

Definition at line 691 of file Descriptors.hpp.

Referenced by BatchToSpaceInferOutputShapeTest(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().

m_Crops

std::vector<std::pair<unsigned int, unsigned int> > m_Crops

The values to crop from the input dimension.

Definition at line 693 of file Descriptors.hpp.

Referenced by BatchToSpaceInferOutputShapeTest(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().

m_DataLayout

DataLayout m_DataLayout

The data layout to be used (NCHW, NHWC).

Definition at line 695 of file Descriptors.hpp.

Referenced by BatchToSpaceInferOutputShapeTest(), BOOST_AUTO_TEST_CASE(), armnn::ClBatchToSpaceNdWorkloadValidate(), TfLiteParser::CreateNetworkFromBinary(), RefBatchToSpaceNdWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), StringifyLayerParameters< BatchToSpaceNdDescriptor >::Serialize(), and SerializerVisitor::VisitBatchToSpaceNdLayer().

---

The documentation for this struct was generated from the following file:

• include/armnn/Descriptors.hpp