ArmNN
 21.02
Reduce.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "BaseIterator.hpp"
9 #include "Decoders.hpp"
10 #include "Encoders.hpp"
11 
12 #include <armnn/Tensor.hpp>
13 
14 namespace armnn
15 {
16 
17 void Reduce(const TensorInfo& inputInfo,
18  const TensorInfo& outputInfo,
19  Decoder<float>& input,
20  Encoder<float>& output,
21  const std::vector<uint32_t> axis,
22  const ReduceOperation reduceOperation);
23 
24 } //namespace armnn
Tensor.hpp
armnn::Reduce
void Reduce(const TensorInfo &inputInfo, const TensorInfo &outputInfo, Decoder< float > &input, Encoder< float > &output, const std::vector< uint32_t > axis, const ReduceOperation reduceOperation)
Definition: Reduce.cpp:71
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
Encoders.hpp
BaseIterator.hpp
armnn::ReduceOperation
ReduceOperation
Definition: Types.hpp:111
Decoders.hpp