diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-09-13 17:20:04 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | e55b40a4d0cc5a82b8f0fd9ffec203ded9f3c63d (patch) | |
| tree | e7736258428837e3889108909d58592937fe71fd /tests/validation | |
| parent | 64f1a908841913049ccc0eb941b5b213290d7bf7 (diff) | |
| download | ComputeLibrary-e55b40a4d0cc5a82b8f0fd9ffec203ded9f3c63d.tar.gz | |
COMPMID-1581: Collapse windows
Change-Id: Iec56c9a96d9736a63f13b65efa33311950f20661
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148572
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: bsgcomp <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation')
| -rw-r--r-- | tests/validation/reference/Col2Im.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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<T> col2im(const SimpleTensor<T> &src, const TensorShape &dst_shape, SimpleTensor<T> 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(); |