20.11
Mean.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
armnn/DescriptorsFwd.hpp
"
9
#include "
armnn/Tensor.hpp
"
10
#include "
BaseIterator.hpp
"
11
12
#include <vector>
13
14
namespace
armnn
15
{
16
void
Mean
(
const
TensorInfo& inputInfo,
17
const
TensorInfo& outputInfo,
18
const
std::vector<unsigned int>& axis,
19
Decoder<float>& input,
20
Encoder<float>& output);
21
}
//namespace armnn
22
Tensor.hpp
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
BaseIterator.hpp
armnn::Mean
void Mean(const armnn::TensorInfo &inputInfo, const armnn::TensorInfo &outputInfo, const std::vector< unsigned int > &axis, Decoder< float > &input, Encoder< float > &output)
Definition:
Mean.cpp:71
DescriptorsFwd.hpp
src
backends
reference
workloads
Mean.hpp
Generated on Wed Nov 25 2020 09:59:14 for ArmNN by
1.8.13