ArmNN
 24.02
Gather.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "armnn/Tensor.hpp"
9 
10 #include "BaseIterator.hpp"
11 #include "Decoders.hpp"
12 #include "Encoders.hpp"
13 
14 namespace armnn
15 {
16 
17 void Gather(const TensorInfo& paramsInfo,
18  const TensorInfo& indicesInfo,
19  const TensorInfo& outputInfo,
20  Decoder<float>& params,
21  const int32_t* indices,
22  Encoder<float>& output,
23  const int32_t = 0);
24 
25 } //namespace armnn
BaseIterator.hpp
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp
armnn::Gather
void Gather(const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< float > &params, const int32_t *indices, Encoder< float > &output, const int32_t axis_int)
Definition: Gather.cpp:15