diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-04-23 16:11:45 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:54 +0000 |
| commit | 764b1af5f41749624c3900fc65c9dad3b1adf98c (patch) | |
| tree | 9d8397de05d86eb40959d24058939c46c36f0d39 /tests/validation/reference/Im2Col.h | |
| parent | 3ca9786fe8ed00ad03963cae6a9eef7bb2fe630e (diff) | |
| download | ComputeLibrary-764b1af5f41749624c3900fc65c9dad3b1adf98c.tar.gz | |
COMPMID-1070: Rewrote im2col nhwc reference
The new reference computes directly the results without the need
of using temporary tensors and perform permutations which is problematic
for big 4k tensors as we get bad_alloc execptions.
Change-Id: I1fb0a495c4e9ca3356de76c9298832db8ace794a
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128683
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/validation/reference/Im2Col.h')
| -rw-r--r-- | tests/validation/reference/Im2Col.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/Im2Col.h b/tests/validation/reference/Im2Col.h index 4fe6ea9acf..5277171a2f 100644 --- a/tests/validation/reference/Im2Col.h +++ b/tests/validation/reference/Im2Col.h @@ -35,7 +35,7 @@ namespace validation namespace reference { template <typename T> -SimpleTensor<T> im2col(const SimpleTensor<T> &src, const TensorShape &dst_shape, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias); +void im2col(const SimpleTensor<T> &src, SimpleTensor<T> &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias); } // namespace reference } // namespace validation } // namespace test |