ArmNN
 24.08
ScatterNd.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2024 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <armnn/Tensor.hpp>
9 #include "Encoders.hpp"
10 #include "Decoders.hpp"
11 #include "armnn/Descriptors.hpp"
12 
13 namespace armnn
14 {
15 // ScatterNd with input tensor
16 void ScatterNd(const TensorInfo& inputInfo,
17  const TensorInfo& indicesInfo,
18  const TensorInfo& updatesInfo,
19  Decoder<float>& input,
20  Decoder<int>& indices,
21  Decoder<float>& updates,
22  Encoder<float>& output,
23  ScatterNdDescriptor descriptor);
24 
25 // ScatterNd without input tensor, only shape provided
26 void ScatterNd(const TensorInfo& indicesInfo,
27  const TensorInfo& updatesInfo,
28  const TensorInfo& shapeInfo,
29  Decoder<int>& indices,
30  Decoder<float>& updates,
31  Decoder<int>& shape,
32  Encoder<float>& output,
33  ScatterNdDescriptor descriptor);
34 } // namespace armnn
Descriptors.hpp
armnn::ScatterNd
void ScatterNd(const TensorInfo &inputInfo, const TensorInfo &indicesInfo, const TensorInfo &updatesInfo, Decoder< float > &input, Decoder< int > &indices, Decoder< float > &updates, Encoder< float > &output, ScatterNdDescriptor descriptor)
Definition: ScatterNd.cpp:41
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp