ArmNN  NotReleased
ElementwiseFunction.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 "BaseIterator.hpp"
9 #include <armnn/Tensor.hpp>
10 
11 namespace armnn
12 {
13 
14 template <typename Functor>
16 {
17  using OutType = typename Functor::result_type;
18  using InType = typename Functor::first_argument_type;
19 
20  ElementwiseBinaryFunction(const TensorShape& inShape0,
21  const TensorShape& inShape1,
22  const TensorShape& outShape,
23  Decoder<InType>& inData0,
24  Decoder<InType>& inData1,
25  Encoder<OutType>& outData);
26 };
27 
28 template <typename Functor>
30 {
31  using OutType = typename Functor::result_type;
32  using InType = typename Functor::argument_type;
33 
35  const TensorShape& outShape,
36  Decoder<InType>& inData,
37  Encoder<OutType>& outData);
38 };
39 
40 } //namespace armnn
typename Functor::argument_type InType
typename Functor::result_type OutType
ElementwiseBinaryFunction(const TensorShape &inShape0, const TensorShape &inShape1, const TensorShape &outShape, Decoder< InType > &inData0, Decoder< InType > &inData1, Encoder< OutType > &outData)
typename Functor::first_argument_type InType
typename Functor::result_type OutType