blob: 38543393ce70aaa5a98f950ffae1a9a6f9379058 [file] [log] [blame]
Jenkinsb3a371b2018-05-23 11:36:53 +01001/*
Jenkins514be652019-02-28 12:25:18 +00002 * Copyright (c) 2018-2019 ARM Limited.
Jenkinsb3a371b2018-05-23 11:36:53 +01003 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24#include "WidthConcatenateLayer.h"
25
Jenkinsb3a371b2018-05-23 11:36:53 +010026#include "tests/validation/Helpers.h"
27
28namespace arm_compute
29{
30namespace test
31{
32namespace validation
33{
34namespace reference
35{
36template <typename T>
Jenkins514be652019-02-28 12:25:18 +000037SimpleTensor<T> widthconcatenate_layer(const std::vector<SimpleTensor<T>> &srcs, SimpleTensor<T> &dst)
Jenkinsb3a371b2018-05-23 11:36:53 +010038{
39 // Create reference
40 std::vector<TensorShape> shapes;
41
42 for(const auto &src : srcs)
43 {
44 shapes.emplace_back(src.shape());
45 }
46
Jenkinsb3a371b2018-05-23 11:36:53 +010047 // Compute reference
48 int width_offset = 0;
49 const int width_out = dst.shape().x();
50
51 // Set output tensor to 0
52 std::fill_n(dst.data(), dst.num_elements(), 0);
53
54 for(const auto &src : srcs)
55 {
56 ARM_COMPUTE_ERROR_ON(width_offset >= width_out);
57
Jenkinsb9abeae2018-11-22 11:58:08 +000058 const int width = src.shape().x();
59 const int height = src.shape().y();
60 const int depth = src.shape().z();
61 const int upper_dims = src.shape().total_size() / (width * height * depth);
Jenkinsb3a371b2018-05-23 11:36:53 +010062
63 const T *src_ptr = src.data();
64 T *dst_ptr = dst.data();
65
Jenkinsb9abeae2018-11-22 11:58:08 +000066 for(int u = 0; u < upper_dims; ++u)
Jenkinsb3a371b2018-05-23 11:36:53 +010067 {
Jenkinsb9abeae2018-11-22 11:58:08 +000068 for(int d = 0; d < depth; ++d)
Jenkinsb3a371b2018-05-23 11:36:53 +010069 {
Jenkinsb9abeae2018-11-22 11:58:08 +000070 for(int r = 0; r < height; ++r)
71 {
72 const int offset = u * height * depth + d * height + r;
Jenkins514be652019-02-28 12:25:18 +000073 if(src.data_type() == DataType::QASYMM8 && src.quantization_info() != dst.quantization_info())
74 {
75 std::transform(src_ptr, src_ptr + width, dst_ptr + width_offset + offset * width_out, [src, dst](T t)
76 {
77 const float dequantized_input = src.quantization_info().dequantize(t);
78 return dst.quantization_info().quantize(dequantized_input, RoundingPolicy::TO_NEAREST_UP);
79 });
80 src_ptr += width;
81 }
82 else
83 {
84 std::copy(src_ptr, src_ptr + width, dst_ptr + width_offset + offset * width_out);
85 src_ptr += width;
86 }
Jenkinsb9abeae2018-11-22 11:58:08 +000087 }
Jenkinsb3a371b2018-05-23 11:36:53 +010088 }
89 }
Jenkinsb3a371b2018-05-23 11:36:53 +010090 width_offset += width;
91 }
92
93 return dst;
94}
95
Jenkins514be652019-02-28 12:25:18 +000096template SimpleTensor<float> widthconcatenate_layer(const std::vector<SimpleTensor<float>> &srcs, SimpleTensor<float> &dst);
97template SimpleTensor<half> widthconcatenate_layer(const std::vector<SimpleTensor<half>> &srcs, SimpleTensor<half> &dst);
98template SimpleTensor<uint8_t> widthconcatenate_layer(const std::vector<SimpleTensor<uint8_t>> &srcs, SimpleTensor<uint8_t> &dst);
Jenkinsb3a371b2018-05-23 11:36:53 +010099} // namespace reference
100} // namespace validation
101} // namespace test
102} // namespace arm_compute