diff options
author | SiCong Li <sicong.li@arm.com> | 2023-03-21 12:00:15 +0000 |
---|---|---|
committer | SiCong Li <sicong.li@arm.com> | 2023-03-27 12:56:26 +0000 |
commit | 5a7d1571a2de24eefc6f1d8d22deeef9f47521ee (patch) | |
tree | 1a9610a60f468619aa54acd4454ace59d83b5b88 /tests/validation/reference/BatchToSpaceLayer.h | |
parent | b531b7549abdd5c10b14b00107ea647591baa430 (diff) | |
download | ComputeLibrary-5a7d1571a2de24eefc6f1d8d22deeef9f47521ee.tar.gz |
Fix BatchToSpaceFixture
* Use a vector to represent the (static) block shape instead of an N-D
Tensor. The previous use of ND Tensor as block shape was wrong, not
adhering to the specification, and non-functional (only first dim was
used anyway).
* The fixture now accepts a static block shape, because the dynamic
case is not properly implemented and will be deprecated for now.
* Fix an assertion error in reference implementation.
Partially resolves COMPMID-5918
Change-Id: I5221e52ccc05e7c1249dec3a42426f954a73729a
Signed-off-by: SiCong Li <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9357
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-by: Omar Al Khatib <omar.alkhatib@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference/BatchToSpaceLayer.h')
-rw-r--r-- | tests/validation/reference/BatchToSpaceLayer.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/BatchToSpaceLayer.h b/tests/validation/reference/BatchToSpaceLayer.h index 18010f1885..a37bfc3373 100644 --- a/tests/validation/reference/BatchToSpaceLayer.h +++ b/tests/validation/reference/BatchToSpaceLayer.h @@ -37,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, const CropInfo &crop_info = CropInfo{}); +SimpleTensor<T> batch_to_space(const SimpleTensor<T> &src, const std::vector<int32_t> &block_shape, const CropInfo &crop_info, const TensorShape &dst_shape); } // namespace reference } // namespace validation } // namespace test |