diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-07-14 15:29:28 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-07-17 11:16:17 +0000 |
commit | e068199254e525176b2c1eaf8420b9ddac3d9011 (patch) | |
tree | 1ef6ab5dcb4311c4c599bd93b2788b9a4c6d271f /tests/validation/reference | |
parent | a084b46835d20fdfe6e590b91b7ca64fba3542df (diff) | |
download | ComputeLibrary-e068199254e525176b2c1eaf8420b9ddac3d9011.tar.gz |
COMPMID-3576: Nightly failure: NEON/PoolingLayer/Float/FP16/MaxUnpooling S10
Extend NEPoolingLayer max pooling to extract indices for FP16
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I5a7c754be353e4c2c5d0ab3794e9427408d0c4fa
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3580
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference')
-rw-r--r-- | tests/validation/reference/MaxUnpoolingLayer.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/validation/reference/MaxUnpoolingLayer.cpp b/tests/validation/reference/MaxUnpoolingLayer.cpp index 134f39726a..880018e99f 100644 --- a/tests/validation/reference/MaxUnpoolingLayer.cpp +++ b/tests/validation/reference/MaxUnpoolingLayer.cpp @@ -99,6 +99,9 @@ SimpleTensor<T> max_unpooling_layer(const SimpleTensor<T> &src, const PoolingLay template SimpleTensor<float> max_unpooling_layer(const SimpleTensor<float> &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> &indices, TensorShape output_shape, DataLayout data_layout); +template SimpleTensor<half> max_unpooling_layer(const SimpleTensor<half> &src, const PoolingLayerInfo &info, + const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> &indices, + TensorShape output_shape, DataLayout data_layout); } // namespace reference } // namespace validation |