From e55b40a4d0cc5a82b8f0fd9ffec203ded9f3c63d Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 13 Sep 2018 17:20:04 +0100 Subject: COMPMID-1581: Collapse windows Change-Id: Iec56c9a96d9736a63f13b65efa33311950f20661 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148572 Reviewed-by: Anthony Barbier Tested-by: bsgcomp --- tests/validation/reference/Col2Im.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/validation/reference/Col2Im.cpp') diff --git a/tests/validation/reference/Col2Im.cpp b/tests/validation/reference/Col2Im.cpp index 90e488f928..53969d4725 100644 --- a/tests/validation/reference/Col2Im.cpp +++ b/tests/validation/reference/Col2Im.cpp @@ -40,7 +40,7 @@ SimpleTensor col2im(const SimpleTensor &src, const TensorShape &dst_shape, SimpleTensor dst{ dst_shape, src.data_type(), 1 }; // Compute reference - const size_t batches = dst_shape[3]; + const size_t batches = dst_shape.total_size() / (dst_shape.x() * dst_shape.y() * dst_shape.z()); const size_t src_width = src.shape().x(); const size_t src_height = src.shape().y(); -- cgit v1.2.1