20.11
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
Tensor.hpp
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
Encoders.hpp
BaseIterator.hpp
armnn::Gather
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)
Definition:
Gather.cpp:17
Decoders.hpp
src
backends
reference
workloads
Gather.hpp
Generated on Wed Nov 25 2020 09:59:14 for ArmNN by
1.8.13