21 const int32_t* indices,
30 unsigned int paramsProduct = 1;
33 paramsProduct = paramsProduct * paramsShape[i];
36 unsigned int outIndex = 0;
43 unsigned int startOffset = indx * paramsProduct;
44 unsigned int endOffset = startOffset + paramsProduct;
46 for (
unsigned int j = startOffset; j < endOffset; ++j)
49 float outputValue = params.
Get();
51 output.
Set(outputValue);
const TensorShape & GetShape() const
virtual void Set(IType right)=0
Copyright (c) 2020 ARM Limited.
void IgnoreUnused(Ts &&...)
virtual IType Get() const =0
void Gather(const TensorInfo ¶msInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< float > ¶ms, const int32_t *indices, Encoder< float > &output, const int32_t axis)
#define ARMNN_ASSERT(COND)
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
unsigned int GetNumDimensions() const
unsigned int GetNumElements() const