|
25.11
|
A ReduceDescriptor for the REDUCE operators. More...
#include <Descriptors.hpp>
Public Member Functions | |
| ReduceDescriptor () | |
| bool | operator== (const ReduceDescriptor &rhs) const |
| Public Member Functions inherited from BaseDescriptor | |
| virtual bool | IsNull () const |
| virtual | ~BaseDescriptor ()=default |
Public Attributes | |
| bool | m_KeepDims |
| if true then output shape has no change. | |
| std::vector< uint32_t > | m_vAxis |
| The indices of the dimensions to reduce. | |
| ReduceOperation | m_ReduceOperation |
| Specifies the reduction operation to execute. | |
A ReduceDescriptor for the REDUCE operators.
Definition at line 1538 of file Descriptors.hpp.
|
inline |
Definition at line 1540 of file Descriptors.hpp.
References m_KeepDims, m_ReduceOperation, m_vAxis, and armnn::Sum.
Referenced by operator==().
|
inline |
Definition at line 1546 of file Descriptors.hpp.
References m_KeepDims, m_ReduceOperation, m_vAxis, and ReduceDescriptor().
| bool m_KeepDims |
if true then output shape has no change.
Definition at line 1554 of file Descriptors.hpp.
Referenced by armnn::ChainReduceLayers(), armnn::ClReduceWorkloadValidate(), ReduceLayer::CreateWorkload(), GetTosaMapping(), armnn::IsMultiAxesReduceSupported(), armnn::NeonReduceWorkloadValidate(), operator==(), ReduceDescriptor(), and StringifyLayerParameters< ReduceDescriptor >::Serialize().
| ReduceOperation m_ReduceOperation |
Specifies the reduction operation to execute.
Definition at line 1558 of file Descriptors.hpp.
Referenced by ConvertReduceToTosaOperator(), armnn::ConvertReductionOperationToAcl(), ReduceLayer::CreateWorkload(), GetTosaMapping(), operator==(), ReduceDescriptor(), and StringifyLayerParameters< ReduceDescriptor >::Serialize().
| std::vector<uint32_t> m_vAxis |
The indices of the dimensions to reduce.
Definition at line 1556 of file Descriptors.hpp.
Referenced by armnn::ChainReduceLayers(), armnn::ClReduceWorkloadValidate(), ConvertReduceToTosaOperator(), ReduceLayer::CreateWorkload(), GetTosaMapping(), armnn::IsMultiAxesReduceSupported(), armnn::NeonReduceWorkloadValidate(), operator==(), ClBackend::OptimizeSubgraphView(), NeonBackend::OptimizeSubgraphView(), ReduceDescriptor(), and StringifyLayerParameters< ReduceDescriptor >::Serialize().