21.05
|
A ReduceDescriptor for the REDUCE operators. More...
#include <Descriptors.hpp>
Public Member Functions | |
ReduceDescriptor () | |
bool | operator== (const ReduceDescriptor &rhs) const |
Public Attributes | |
bool | m_KeepDims |
if true then output shape has no change. More... | |
std::vector< uint32_t > | m_vAxis |
The indices of the dimensions to reduce. More... | |
ReduceOperation | m_ReduceOperation |
Specifies the reduction operation to execute. More... | |
A ReduceDescriptor for the REDUCE operators.
Definition at line 1312 of file Descriptors.hpp.
|
inline |
Definition at line 1314 of file Descriptors.hpp.
|
inline |
Definition at line 1320 of file Descriptors.hpp.
References ReduceDescriptor::m_KeepDims, ReduceDescriptor::m_ReduceOperation, and ReduceDescriptor::m_vAxis.
bool m_KeepDims |
if true then output shape has no change.
Definition at line 1328 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::ComputeWrappedIndex(), ReduceLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), ReduceDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< ReduceDescriptor >::Serialize(), and ReduceLayer::ValidateTensorShapesFromInputs().
ReduceOperation m_ReduceOperation |
Specifies the reduction operation to execute.
Definition at line 1332 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::ComputeWrappedIndex(), armnn::ConvertReductionOperationToAcl(), ReduceLayer::CreateWorkload(), RefReduceWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), ReduceDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), and StringifyLayerParameters< ReduceDescriptor >::Serialize().
std::vector<uint32_t> m_vAxis |
The indices of the dimensions to reduce.
Definition at line 1330 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), ReduceLayer::Clone(), armnnTfLiteParser::ComputeWrappedIndex(), ReduceLayer::CreateWorkload(), RefReduceWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), ReduceDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< ReduceDescriptor >::Serialize(), and ReduceLayer::ValidateTensorShapesFromInputs().