21.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 951 of file Descriptors.hpp.
|
inline |
Definition at line 953 of file Descriptors.hpp.
|
inline |
Definition at line 958 of file Descriptors.hpp.
|
inline |
Definition at line 963 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 969 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), MeanLayer::Clone(), MeanLayer::CreateWorkload(), RefMeanWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseMean(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanLayer::ValidateTensorShapesFromInputs().
bool m_KeepDims |
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.
Definition at line 971 of file Descriptors.hpp.
Referenced by MeanLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseMean(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanLayer::ValidateTensorShapesFromInputs().