21.08
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:123
Decoders.hpp
src
backends
reference
workloads
Reduce.hpp
Generated on Tue Aug 24 2021 16:18:44 for ArmNN by
1.8.13