xref: /aosp_15_r20/external/ComputeLibrary/tests/datasets/ConvolutionLayerDataset.h (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2017-2018 Arm Limited.
3  *
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 #ifndef ARM_COMPUTE_TEST_CONVOLUTION_LAYER_DATASET
25 #define ARM_COMPUTE_TEST_CONVOLUTION_LAYER_DATASET
26 
27 #include "utils/TypePrinter.h"
28 
29 #include "arm_compute/core/TensorShape.h"
30 #include "arm_compute/core/Types.h"
31 
32 namespace arm_compute
33 {
34 namespace test
35 {
36 namespace datasets
37 {
38 class ConvolutionLayerDataset
39 {
40 public:
41     using type = std::tuple<TensorShape, TensorShape, TensorShape, TensorShape, PadStrideInfo, Size2D>;
42 
43     struct iterator
44     {
iteratoriterator45         iterator(std::vector<TensorShape>::const_iterator   src_it,
46                  std::vector<TensorShape>::const_iterator   weights_it,
47                  std::vector<TensorShape>::const_iterator   biases_it,
48                  std::vector<TensorShape>::const_iterator   dst_it,
49                  std::vector<PadStrideInfo>::const_iterator infos_it,
50                  std::vector<Size2D>::const_iterator        dilation_it)
51             : _src_it{ std::move(src_it) },
52               _weights_it{ std::move(weights_it) },
53               _biases_it{ std::move(biases_it) },
54               _dst_it{ std::move(dst_it) },
55               _infos_it{ std::move(infos_it) },
56               _dilation_it{ std::move(dilation_it) }
57         {
58         }
59 
descriptioniterator60         std::string description() const
61         {
62             std::stringstream description;
63             description << "In=" << *_src_it << ":";
64             description << "Weights=" << *_weights_it << ":";
65             description << "Biases=" << *_biases_it << ":";
66             description << "Out=" << *_dst_it << ":";
67             description << "Info=" << *_infos_it << ":";
68             description << "Dilation=" << *_dilation_it;
69             return description.str();
70         }
71 
72         ConvolutionLayerDataset::type operator*() const
73         {
74             return std::make_tuple(*_src_it, *_weights_it, *_biases_it, *_dst_it, *_infos_it, *_dilation_it);
75         }
76 
77         iterator &operator++()
78         {
79             ++_src_it;
80             ++_weights_it;
81             ++_biases_it;
82             ++_dst_it;
83             ++_infos_it;
84             ++_dilation_it;
85 
86             return *this;
87         }
88 
89     private:
90         std::vector<TensorShape>::const_iterator   _src_it;
91         std::vector<TensorShape>::const_iterator   _weights_it;
92         std::vector<TensorShape>::const_iterator   _biases_it;
93         std::vector<TensorShape>::const_iterator   _dst_it;
94         std::vector<PadStrideInfo>::const_iterator _infos_it;
95         std::vector<Size2D>::const_iterator        _dilation_it;
96     };
97 
begin()98     iterator begin() const
99     {
100         return iterator(_src_shapes.begin(), _weight_shapes.begin(), _bias_shapes.begin(), _dst_shapes.begin(), _infos.begin(), _dilations.begin());
101     }
102 
size()103     int size() const
104     {
105         return std::min(_src_shapes.size(), std::min(_weight_shapes.size(), std::min(_bias_shapes.size(), std::min(_dst_shapes.size(), std::min(_infos.size(), _dilations.size())))));
106     }
107 
108     void add_config(TensorShape src, TensorShape weights, TensorShape biases, TensorShape dst, PadStrideInfo info, Size2D dilation = Size2D(1U, 1U))
109     {
110         _src_shapes.emplace_back(std::move(src));
111         _weight_shapes.emplace_back(std::move(weights));
112         _bias_shapes.emplace_back(std::move(biases));
113         _dst_shapes.emplace_back(std::move(dst));
114         _infos.emplace_back(std::move(info));
115         _dilations.emplace_back(std::move(dilation));
116     }
117 
118 protected:
119     ConvolutionLayerDataset()                           = default;
120     ConvolutionLayerDataset(ConvolutionLayerDataset &&) = default;
121 
122 private:
123     std::vector<TensorShape>   _src_shapes{};
124     std::vector<TensorShape>   _weight_shapes{};
125     std::vector<TensorShape>   _bias_shapes{};
126     std::vector<TensorShape>   _dst_shapes{};
127     std::vector<PadStrideInfo> _infos{};
128     std::vector<Size2D>        _dilations{};
129 };
130 } // namespace datasets
131 } // namespace test
132 } // namespace arm_compute
133 #endif /* ARM_COMPUTE_TEST_CONVOLUTION_LAYER_DATASET */
134