diff options
author | Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-06-28 14:09:53 +0100 |
---|---|---|
committer | VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-06-28 14:15:30 +0000 |
commit | 338595bca8ab60492f10626860acb1ab3722b1ce (patch) | |
tree | 03504ec3a2973e30c80f9bf56b77b4a4c7c9d83c /tests/validation/reference/ConcatenateLayer.cpp | |
parent | 7026b303d636e7639f8877ae8d5eff54f39c1121 (diff) | |
download | ComputeLibrary-338595bca8ab60492f10626860acb1ab3722b1ce.tar.gz |
COMPMID-2234 : Add support for axis 3 in NE/CLConcatenateLayer
Change-Id: Ic86f89ece3afe72809bc69c6de6fee7d21daa1d4
Signed-off-by: Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1440
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/reference/ConcatenateLayer.cpp')
-rw-r--r-- | tests/validation/reference/ConcatenateLayer.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/validation/reference/ConcatenateLayer.cpp b/tests/validation/reference/ConcatenateLayer.cpp index 6c90d74a0f..aa74ca2474 100644 --- a/tests/validation/reference/ConcatenateLayer.cpp +++ b/tests/validation/reference/ConcatenateLayer.cpp @@ -127,6 +127,16 @@ SimpleTensor<T> concatenate_layer(std::vector<SimpleTensor<T>> &srcs, SimpleTens dst = reference::permute<T>(dst, PermutationVector(2U, 1U, 0U)); return reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(2U, 1U, 0U)); } + case 3: + { + for(auto &t : srcs) + { + t = reference::permute<T>(t, PermutationVector(3U, 2U, 1U, 0U)); + } + dst = reference::permute<T>(dst, PermutationVector(3U, 2U, 1U, 0U)); + auto ret = reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(3U, 2U, 1U, 0U)); + return ret; + } default: { ARM_COMPUTE_ERROR("Not supported"); |