21.05
|
A GatherDescriptor for the GatherLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
GatherDescriptor () | |
GatherDescriptor (int32_t axis) | |
bool | operator== (const GatherDescriptor &rhs) const |
Public Attributes | |
int32_t | m_Axis |
The axis in params to gather indices from. More... | |
A GatherDescriptor for the GatherLayer.
Definition at line 750 of file Descriptors.hpp.
|
inline |
Definition at line 752 of file Descriptors.hpp.
|
inline |
Definition at line 756 of file Descriptors.hpp.
|
inline |
Definition at line 760 of file Descriptors.hpp.
References GatherDescriptor::m_Axis.
int32_t m_Axis |
The axis in params to gather indices from.
Definition at line 766 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), ClGatherWorkload::ClGatherWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), RefGatherWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsGatherSupported(), NeonGatherWorkload::NeonGatherWorkload(), GatherDescriptor::operator==(), and GatherLayer::ValidateTensorShapesFromInputs().