ArmNN
 25.11
Loading...
Searching...
No Matches
Gather.hpp File Reference
#include "armnn/Tensor.hpp"
#include "BaseIterator.hpp"
#include "Decoders.hpp"
#include "Encoders.hpp"
Include dependency graph for Gather.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  armnn
 Copyright (c) 2021 ARM Limited and Contributors.

Functions

template<typename I, typename O>
void Gather (const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< I > &params, const int32_t *indices, Encoder< O > &output, const int32_t axis_int)