20.02
|
A MeanDescriptor for the MeanLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
MeanDescriptor () | |
MeanDescriptor (const std::vector< unsigned int > &axis, bool keepDims) | |
bool | operator== (const MeanDescriptor &rhs) const |
Public Attributes | |
std::vector< unsigned int > | m_Axis |
Values for the dimensions to reduce. More... | |
bool | m_KeepDims |
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept. More... | |
A MeanDescriptor for the MeanLayer.
Definition at line 877 of file Descriptors.hpp.
|
inline |
Definition at line 879 of file Descriptors.hpp.
|
inline |
Definition at line 884 of file Descriptors.hpp.
|
inline |
Definition at line 889 of file Descriptors.hpp.
References MeanDescriptor::m_Axis, and MeanDescriptor::m_KeepDims.
std::vector<unsigned int> m_Axis |
Values for the dimensions to reduce.
Definition at line 895 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), MeanLayer::Clone(), MeanLayer::CreateWorkload(), RefMeanWorkload::Execute(), Deserializer::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< MeanDescriptor >::Serialize(), MeanLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitMeanLayer().
bool m_KeepDims |
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.
Definition at line 897 of file Descriptors.hpp.
Referenced by MeanLayer::CreateWorkload(), Deserializer::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< MeanDescriptor >::Serialize(), MeanLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitMeanLayer().