diff options
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"); |