diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-04-03 19:48:54 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-04-05 10:24:49 +0000 |
commit | 9f15c5e838085294ce391d3bdcf5ec7752650ea8 (patch) | |
tree | c3abb2d06998ffb0a1bf3cb7da52359d6c2d562d /tests/validation/reference/ConcatenateLayer.cpp | |
parent | 7191aaa184992ca1fa8bdbbe5a9d8f9cd093ad5b (diff) | |
download | ComputeLibrary-9f15c5e838085294ce391d3bdcf5ec7752650ea8.tar.gz |
COMPMID-2062 Rework NEON ConcatenateLayer
Change-Id: I2703c99f651e4f06c4e44e39a85a8a8d201c5362
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/944
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'tests/validation/reference/ConcatenateLayer.cpp')
-rw-r--r-- | tests/validation/reference/ConcatenateLayer.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/validation/reference/ConcatenateLayer.cpp b/tests/validation/reference/ConcatenateLayer.cpp index 1440878829..93b1768ebd 100644 --- a/tests/validation/reference/ConcatenateLayer.cpp +++ b/tests/validation/reference/ConcatenateLayer.cpp @@ -114,6 +114,15 @@ SimpleTensor<T> concatenate_layer(std::vector<SimpleTensor<T>> &srcs, SimpleTens dst = reference::permute<T>(dst, PermutationVector(1U, 0U)); return reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(1U, 0U)); } + case Window::DimZ: + { + for(auto &t : srcs) + { + t = reference::permute<T>(t, PermutationVector(2U, 1U, 0U)); + } + dst = reference::permute<T>(dst, PermutationVector(2U, 1U, 0U)); + return reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(2U, 1U, 0U)); + } default: { ARM_COMPUTE_ERROR("Not supported"); |