diff options
author | SiCong Li <sicong.li@arm.com> | 2023-03-13 15:02:23 +0000 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2023-03-14 15:38:29 +0000 |
commit | 4ceb453b00185ded5ddbaf83d40eadeb2ed28ec4 (patch) | |
tree | 13d56b417d5c2b186bde627f4f5d0f05b7228a53 /tests/validation/reference/BatchToSpaceLayer.h | |
parent | aaa9da1efa83911c7a67d50811ad669a92a7d12f (diff) | |
download | ComputeLibrary-4ceb453b00185ded5ddbaf83d40eadeb2ed28ec4.tar.gz |
Add CropInfo to BatchToSpace reference and fixture
Partially resolves COMPMID-5918, COMPMID-5865
Signed-off-by: SiCong Li <sicong.li@arm.com>
Change-Id: Ib3b01e7dc1c944184a4c038045bf0469fbb9ff45
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9321
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference/BatchToSpaceLayer.h')
-rw-r--r-- | tests/validation/reference/BatchToSpaceLayer.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/validation/reference/BatchToSpaceLayer.h b/tests/validation/reference/BatchToSpaceLayer.h index 52556cb53f..18010f1885 100644 --- a/tests/validation/reference/BatchToSpaceLayer.h +++ b/tests/validation/reference/BatchToSpaceLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 Arm Limited. + * Copyright (c) 2018-2019, 2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,6 +24,7 @@ #ifndef ARM_COMPUTE_TEST_BATCH_TO_SPACE_LAYER_H #define ARM_COMPUTE_TEST_BATCH_TO_SPACE_LAYER_H +#include "arm_compute/core/Types.h" #include "tests/SimpleTensor.h" #include "tests/validation/Helpers.h" @@ -36,7 +37,7 @@ namespace validation namespace reference { template <typename T> -SimpleTensor<T> batch_to_space(const SimpleTensor<T> &src, const SimpleTensor<int32_t> &block_shape, const TensorShape &dst_shape); +SimpleTensor<T> batch_to_space(const SimpleTensor<T> &src, const SimpleTensor<int32_t> &block_shape, const TensorShape &dst_shape, const CropInfo &crop_info = CropInfo{}); } // namespace reference } // namespace validation } // namespace test |