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 | |
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')
-rw-r--r-- | tests/validation/NEON/MaxUnpoolingLayer.cpp | 8 | ||||
-rw-r--r-- | tests/validation/reference/MaxUnpoolingLayer.cpp | 3 |
2 files changed, 7 insertions, 4 deletions
diff --git a/tests/validation/NEON/MaxUnpoolingLayer.cpp b/tests/validation/NEON/MaxUnpoolingLayer.cpp index e44021d858..a33ec2885e 100644 --- a/tests/validation/NEON/MaxUnpoolingLayer.cpp +++ b/tests/validation/NEON/MaxUnpoolingLayer.cpp @@ -63,11 +63,11 @@ FIXTURE_DATA_TEST_CASE(MaxUnpooling, NEMaxUnpoolingLayerFixture<float>, framewor TEST_SUITE_END() // FP32 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC TEST_SUITE(FP16) -FIXTURE_DATA_TEST_CASE(MaxUnpooling, NEMaxUnpoolingLayerFixture<float>, framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallShapes(), combine(PoolingLayerIndicesDatasetFPSmall, - framework::dataset::make("DataType", DataType::F16))), - framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC }) +FIXTURE_DATA_TEST_CASE(MaxUnpooling, NEMaxUnpoolingLayerFixture<half>, framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallShapes(), combine(PoolingLayerIndicesDatasetFPSmall, + framework::dataset::make("DataType", DataType::F16))), + framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC }) - )) + )) { // Validate output validate(Accessor(_target), _reference); 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 |