diff options
Diffstat (limited to 'src/backends/reference/workloads/ElementwiseFunction.hpp')
-rw-r--r-- | src/backends/reference/workloads/ElementwiseFunction.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/ElementwiseFunction.hpp b/src/backends/reference/workloads/ElementwiseFunction.hpp index 0ac136466c..8099f3279a 100644 --- a/src/backends/reference/workloads/ElementwiseFunction.hpp +++ b/src/backends/reference/workloads/ElementwiseFunction.hpp @@ -10,15 +10,15 @@ namespace armnn { -template <typename Functor> +template <typename Functor, typename dataTypeInput, typename dataTypeOutput> struct ElementwiseFunction { ElementwiseFunction(const TensorShape& inShape0, const TensorShape& inShape1, const TensorShape& outShape, - const float* inData0, - const float* inData1, - float* outData); + const dataTypeInput* inData0, + const dataTypeInput* inData1, + dataTypeOutput* outData); }; } //namespace armnn |