diff options
| author | SiCongLi <sicong.li@arm.com> | 2021-12-22 15:37:20 +0000 |
|---|---|---|
| committer | SiCong Li <sicong.li@arm.com> | 2021-12-24 15:59:25 +0000 |
| commit | cb86956e1972be4b2ddbaacaa23a0d21185f8ccb (patch) | |
| tree | 69a51b7b8a8121756044fa60657d05d534fb6e02 /tests/validation/fixtures/PoolingLayerFixture.h | |
| parent | b2eba7f307d5ae634ff41bd88d5bd1659466d82d (diff) | |
| download | ComputeLibrary-cb86956e1972be4b2ddbaacaa23a0d21185f8ccb.tar.gz | |
Fix test validation method
* Allow non-finite values to be equal (inf == inf, -inf == -inf) in
validate
* Fix SpecialPoolingLayerValidationFixture
Partially resolves COMPMID-4998
Change-Id: I3fba1ccee74c1af419a3b6088ddac68c79aa243a
Signed-off-by: SiCongLi <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6856
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/fixtures/PoolingLayerFixture.h')
| -rw-r--r-- | tests/validation/fixtures/PoolingLayerFixture.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/fixtures/PoolingLayerFixture.h b/tests/validation/fixtures/PoolingLayerFixture.h index ec4e9f80dd..6e9edfbb5d 100644 --- a/tests/validation/fixtures/PoolingLayerFixture.h +++ b/tests/validation/fixtures/PoolingLayerFixture.h @@ -213,7 +213,7 @@ public: template <typename...> void setup(TensorShape src_shape, PoolingLayerInfo pool_info, DataType data_type) { - PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(src_shape, pool_info, data_type, DataLayout::NCHW); + PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(src_shape, pool_info, data_type, pool_info.data_layout); } }; |