diff options
Diffstat (limited to 'tests/validation/CPP/ReshapeLayer.cpp')
| -rw-r--r-- | tests/validation/CPP/ReshapeLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/validation/CPP/ReshapeLayer.cpp b/tests/validation/CPP/ReshapeLayer.cpp index cc7f15e4b1..42f06e4f5a 100644 --- a/tests/validation/CPP/ReshapeLayer.cpp +++ b/tests/validation/CPP/ReshapeLayer.cpp @@ -23,7 +23,7 @@ */ #include "ReshapeLayer.h" -#include "tests/validation/half.h" +#include "arm_compute/core/Types.h" namespace arm_compute { @@ -49,7 +49,7 @@ template SimpleTensor<uint16_t> reshape_layer(const SimpleTensor<uint16_t> &src, template SimpleTensor<int16_t> reshape_layer(const SimpleTensor<int16_t> &src, const TensorShape &output_shape); template SimpleTensor<uint32_t> reshape_layer(const SimpleTensor<uint32_t> &src, const TensorShape &output_shape); template SimpleTensor<int32_t> reshape_layer(const SimpleTensor<int32_t> &src, const TensorShape &output_shape); -template SimpleTensor<half_float::half> reshape_layer(const SimpleTensor<half_float::half> &src, const TensorShape &output_shape); +template SimpleTensor<half> reshape_layer(const SimpleTensor<half> &src, const TensorShape &output_shape); template SimpleTensor<float> reshape_layer(const SimpleTensor<float> &src, const TensorShape &output_shape); } // namespace reference } // namespace validation |