diff options
| author | David Beck <david.beck@arm.com> | 2018-09-06 16:46:34 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:29 +0100 |
| commit | f195f03e095a5c4dc6880be11af64cab83b5c94b (patch) | |
| tree | d1c6d7d46ed70b915772bd50c5074d13443d9bca /src/armnn/backends/RefWorkloads/Subtraction.hpp | |
| parent | c2044fe9d26a8b6afca48aee04bd5d29f8e27b8d (diff) | |
| download | armnn-f195f03e095a5c4dc6880be11af64cab83b5c94b.tar.gz | |
IVGCVSW-1803 : add Ref Subtraction layer
Change-Id: I4c019d626f9369245eca6d549bbe7a28e141f198
Diffstat (limited to 'src/armnn/backends/RefWorkloads/Subtraction.hpp')
| -rw-r--r-- | src/armnn/backends/RefWorkloads/Subtraction.hpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/armnn/backends/RefWorkloads/Subtraction.hpp b/src/armnn/backends/RefWorkloads/Subtraction.hpp new file mode 100644 index 0000000000..3956797185 --- /dev/null +++ b/src/armnn/backends/RefWorkloads/Subtraction.hpp @@ -0,0 +1,20 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <armnn/Tensor.hpp> + +namespace armnn +{ + +void Subtraction(const TensorShape& inShape0, + const TensorShape& inShape1, + const TensorShape& outShape, + const float* inData0, + const float* inData1, + float* outData); + +} //namespace armnn |