diff options
| author | narpra01 <narumol.prangnawarat@arm.com> | 2018-09-28 11:07:51 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
| commit | 1e4c31dafb1c8984a126fa1d211ed8f9eedaf7cc (patch) | |
| tree | 006e40b3bbfdc4a202cdada8fa9afec0dd8fffae /src/backends/reference/workloads/Mean.hpp | |
| parent | 33cea4db0b2729c5dbd50f9c0985578c60baffdd (diff) | |
| download | armnn-1e4c31dafb1c8984a126fa1d211ed8f9eedaf7cc.tar.gz | |
IVGCVSW-1812 Adding Ref implementation and tests of MeanWorkloads
Change-Id: I6fb15c407024e3b91d5abf4513f8090be5821760
Diffstat (limited to 'src/backends/reference/workloads/Mean.hpp')
| -rw-r--r-- | src/backends/reference/workloads/Mean.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/Mean.hpp b/src/backends/reference/workloads/Mean.hpp new file mode 100644 index 0000000000..38c2e39653 --- /dev/null +++ b/src/backends/reference/workloads/Mean.hpp @@ -0,0 +1,21 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "armnn/DescriptorsFwd.hpp" +#include "armnn/Tensor.hpp" + +#include <vector> + +namespace armnn +{ +void Mean(const TensorInfo& inputInfo, + const TensorInfo& outputInfo, + const std::vector<unsigned int>& axis, + const float* inputData, + float* outputData); +} //namespace armnn + |