xref: /aosp_15_r20/external/ComputeLibrary/tests/datasets/dynamic_fusion/PoolingLayerDataset.h (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1*c217d954SCole Faust /*
2*c217d954SCole Faust  * Copyright (c) 2023 Arm Limited.
3*c217d954SCole Faust  *
4*c217d954SCole Faust  * SPDX-License-Identifier: MIT
5*c217d954SCole Faust  *
6*c217d954SCole Faust  * Permission is hereby granted, free of charge, to any person obtaining a copy
7*c217d954SCole Faust  * of this software and associated documentation files (the "Software"), to
8*c217d954SCole Faust  * deal in the Software without restriction, including without limitation the
9*c217d954SCole Faust  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10*c217d954SCole Faust  * sell copies of the Software, and to permit persons to whom the Software is
11*c217d954SCole Faust  * furnished to do so, subject to the following conditions:
12*c217d954SCole Faust  *
13*c217d954SCole Faust  * The above copyright notice and this permission notice shall be included in all
14*c217d954SCole Faust  * copies or substantial portions of the Software.
15*c217d954SCole Faust  *
16*c217d954SCole Faust  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17*c217d954SCole Faust  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18*c217d954SCole Faust  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19*c217d954SCole Faust  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20*c217d954SCole Faust  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21*c217d954SCole Faust  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22*c217d954SCole Faust  * SOFTWARE.
23*c217d954SCole Faust  */
24*c217d954SCole Faust #include "arm_compute/core/TensorShape.h"
25*c217d954SCole Faust #include "arm_compute/core/Types.h"
26*c217d954SCole Faust #include "utils/TypePrinter.h"
27*c217d954SCole Faust #include "arm_compute/dynamic_fusion/sketch/attributes/Pool2dAttributes.h"
28*c217d954SCole Faust 
29*c217d954SCole Faust 
30*c217d954SCole Faust using Pool2dAttributes = arm_compute::experimental::dynamic_fusion::Pool2dAttributes;
31*c217d954SCole Faust 
32*c217d954SCole Faust namespace arm_compute
33*c217d954SCole Faust {
34*c217d954SCole Faust namespace test
35*c217d954SCole Faust {
36*c217d954SCole Faust namespace datasets
37*c217d954SCole Faust {
38*c217d954SCole Faust 
39*c217d954SCole Faust class DynamicFusionPoolingLayerDataset
40*c217d954SCole Faust {
41*c217d954SCole Faust public:
42*c217d954SCole Faust     using type = std::tuple<TensorShape, Pool2dAttributes>;
43*c217d954SCole Faust 
44*c217d954SCole Faust     struct iterator
45*c217d954SCole Faust     {
iteratoriterator46*c217d954SCole Faust         iterator(std::vector<TensorShape>::const_iterator      src_it,
47*c217d954SCole Faust                  std::vector<Pool2dAttributes>::const_iterator infos_it)
48*c217d954SCole Faust             : _src_it{ std::move(src_it) },
49*c217d954SCole Faust               _infos_it{ std::move(infos_it) }
50*c217d954SCole Faust         {
51*c217d954SCole Faust         }
52*c217d954SCole Faust 
descriptioniterator53*c217d954SCole Faust         std::string description() const
54*c217d954SCole Faust         {
55*c217d954SCole Faust             std::stringstream description;
56*c217d954SCole Faust             description << "In=" << *_src_it << ":";
57*c217d954SCole Faust             description << "Info=" << *_infos_it << ":";
58*c217d954SCole Faust             return description.str();
59*c217d954SCole Faust         }
60*c217d954SCole Faust 
61*c217d954SCole Faust         DynamicFusionPoolingLayerDataset::type operator*() const
62*c217d954SCole Faust         {
63*c217d954SCole Faust             return std::make_tuple(*_src_it, *_infos_it);
64*c217d954SCole Faust         }
65*c217d954SCole Faust 
66*c217d954SCole Faust         iterator &operator++()
67*c217d954SCole Faust         {
68*c217d954SCole Faust             ++_src_it;
69*c217d954SCole Faust             ++_infos_it;
70*c217d954SCole Faust 
71*c217d954SCole Faust             return *this;
72*c217d954SCole Faust         }
73*c217d954SCole Faust 
74*c217d954SCole Faust     private:
75*c217d954SCole Faust         std::vector<TensorShape>::const_iterator      _src_it;
76*c217d954SCole Faust         std::vector<Pool2dAttributes>::const_iterator _infos_it;
77*c217d954SCole Faust     };
78*c217d954SCole Faust 
begin()79*c217d954SCole Faust     iterator begin() const
80*c217d954SCole Faust     {
81*c217d954SCole Faust         return iterator(_src_shapes.begin(), _infos.begin());
82*c217d954SCole Faust     }
83*c217d954SCole Faust 
size()84*c217d954SCole Faust     int size() const
85*c217d954SCole Faust     {
86*c217d954SCole Faust         return std::min(_src_shapes.size(), _infos.size());
87*c217d954SCole Faust     }
88*c217d954SCole Faust 
add_config(TensorShape src,Pool2dAttributes info)89*c217d954SCole Faust     void add_config(TensorShape src, Pool2dAttributes info)
90*c217d954SCole Faust     {
91*c217d954SCole Faust         _src_shapes.emplace_back(std::move(src));
92*c217d954SCole Faust         _infos.emplace_back(std::move(info));
93*c217d954SCole Faust     }
94*c217d954SCole Faust 
95*c217d954SCole Faust protected:
96*c217d954SCole Faust     DynamicFusionPoolingLayerDataset()                       = default;
97*c217d954SCole Faust     DynamicFusionPoolingLayerDataset(DynamicFusionPoolingLayerDataset &&) = default;
98*c217d954SCole Faust 
99*c217d954SCole Faust private:
100*c217d954SCole Faust     std::vector<TensorShape>      _src_shapes{};
101*c217d954SCole Faust     std::vector<Pool2dAttributes> _infos{};
102*c217d954SCole Faust };
103*c217d954SCole Faust 
104*c217d954SCole Faust // Special pooling dataset
105*c217d954SCole Faust class PoolingLayerDatasetSpecialDynamicFusion final : public DynamicFusionPoolingLayerDataset
106*c217d954SCole Faust {
107*c217d954SCole Faust public:
PoolingLayerDatasetSpecialDynamicFusion()108*c217d954SCole Faust     PoolingLayerDatasetSpecialDynamicFusion()
109*c217d954SCole Faust     {
110*c217d954SCole Faust         // NCHW DataLayout
111*c217d954SCole Faust         // Special cases
112*c217d954SCole Faust         add_config(TensorShape(2U, 3U, 4U, 1U), Pool2dAttributes().pool_type(PoolingType::AVG).pool_size(Size2D(2,2)).stride(Size2D(3,3)));
113*c217d954SCole Faust         add_config(TensorShape(60U, 52U, 3U, 2U), Pool2dAttributes().pool_type(PoolingType::AVG).pool_size(Size2D(100,100)).stride(Size2D(5,5)).pad(Padding2D(50,50,50,50)));
114*c217d954SCole Faust         // Asymmetric padding
115*c217d954SCole Faust         add_config(TensorShape(112U, 112U, 32U), Pool2dAttributes().pool_type(PoolingType::MAX).pool_size(Size2D(3,3)).pad(Padding2D(0,1,0,1)).stride(Size2D(2,2)));
116*c217d954SCole Faust         add_config(TensorShape(14U, 14U, 832U), Pool2dAttributes().pool_type(PoolingType::MAX).pool_size(Size2D(2,2)).stride(Size2D(1,1)).pad(Padding2D(0,0,0,0)));
117*c217d954SCole Faust 
118*c217d954SCole Faust     }
119*c217d954SCole Faust };
120*c217d954SCole Faust } // namespace datasets
121*c217d954SCole Faust } // namespace test
122*c217d954SCole Faust } // namespace arm_compute