21.05
|
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 959 of file Descriptors.hpp.
|
inline |
Definition at line 961 of file Descriptors.hpp.
|
inline |
Definition at line 966 of file Descriptors.hpp.
|
inline |
Definition at line 971 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 977 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), MeanLayer::Clone(), MeanLayer::CreateWorkload(), RefMeanWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), 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 979 of file Descriptors.hpp.
Referenced by MeanLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanLayer::ValidateTensorShapesFromInputs().