24.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 Member Functions inherited from BaseDescriptor | |
virtual bool | IsNull () const |
virtual | ~BaseDescriptor ()=default |
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 1172 of file Descriptors.hpp.
|
inline |
Definition at line 1174 of file Descriptors.hpp.
|
inline |
Definition at line 1179 of file Descriptors.hpp.
|
inline |
Definition at line 1184 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 1190 of file Descriptors.hpp.
Referenced by MeanLayer::Clone(), MeanLayer::CreateWorkload(), MeanLayer::InferOutputShapes(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanQueueDescriptor::Validate().
bool m_KeepDims |
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.
Definition at line 1192 of file Descriptors.hpp.
Referenced by MeanLayer::Clone(), MeanLayer::CreateWorkload(), MeanLayer::InferOutputShapes(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanQueueDescriptor::Validate().