tests/datasets/FullyConnectedLayerDataset.h

   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_FULLYCONNECTED_LAYER_DATASET
  25 #define ARM_COMPUTE_TEST_FULLYCONNECTED_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 FullyConnectedLayerDataset
  39 {
  40 public:
  41     using type = std::tuple<TensorShape, TensorShape, TensorShape, TensorShape>;
  42
  43     struct iterator
  44     {
  45         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             : _src_it{ std::move(src_it) },
  50               _weights_it{ std::move(weights_it) },
  51               _biases_it{ std::move(biases_it) },
  52               _dst_it{ std::move(dst_it) }
  53         {
  54         }
  55
  56         std::string description() const
  57         {
  58             std::stringstream description;
  59             description << "In=" << *_src_it << ":";
  60             description << "Weights=" << *_weights_it << ":";
  61             description << "Biases=" << *_biases_it << ":";
  62             description << "Out=" << *_dst_it;
  63             return description.str();
  64         }
  65
  66         FullyConnectedLayerDataset::type operator*() const
  67         {
  68             return std::make_tuple(*_src_it, *_weights_it, *_biases_it, *_dst_it);
  69         }
  70
  71         iterator &operator++()
  72         {
  73             ++_src_it;
  74             ++_weights_it;
  75             ++_biases_it;
  76             ++_dst_it;
  77
  78             return *this;
  79         }
  80
  81     private:
  82         std::vector<TensorShape>::const_iterator _src_it;
  83         std::vector<TensorShape>::const_iterator _weights_it;
  84         std::vector<TensorShape>::const_iterator _biases_it;
  85         std::vector<TensorShape>::const_iterator _dst_it;
  86     };
  87
  88     iterator begin() const
  89     {
  90         return iterator(_src_shapes.begin(), _weight_shapes.begin(), _bias_shapes.begin(), _dst_shapes.begin());
  91     }
  92
  93     int size() const
  94     {
  95         return std::min(_src_shapes.size(), std::min(_weight_shapes.size(), std::min(_bias_shapes.size(), _dst_shapes.size())));
  96     }
  97
  98     void add_config(TensorShape src, TensorShape weights, TensorShape biases, TensorShape dst)
  99     {
 100         _src_shapes.emplace_back(std::move(src));
 101         _weight_shapes.emplace_back(std::move(weights));
 102         _bias_shapes.emplace_back(std::move(biases));
 103         _dst_shapes.emplace_back(std::move(dst));
 104     }
 105
 106 protected:
 107     FullyConnectedLayerDataset()                              = default;
 108     FullyConnectedLayerDataset(FullyConnectedLayerDataset &&) = default;
 109
 110 private:
 111     std::vector<TensorShape> _src_shapes{};
 112     std::vector<TensorShape> _weight_shapes{};
 113     std::vector<TensorShape> _bias_shapes{};
 114     std::vector<TensorShape> _dst_shapes{};
 115 };
 116
 117 class TinyFullyConnectedLayerDataset final : public FullyConnectedLayerDataset
 118 {
 119 public:
 120     TinyFullyConnectedLayerDataset()
 121     {
 122         // Conv -> FC
 123         add_config(TensorShape(8U, 1U, 1U), TensorShape(8U, 16U), TensorShape(16U), TensorShape(16U));
 124         // Conv -> FC
 125         add_config(TensorShape(1U, 1U, 1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U));
 126         // FC -> FC
 127         add_config(TensorShape(1U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U));
 128         // FC -> FC (batched)
 129         add_config(TensorShape(1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U));
 130     }
 131 };
 132 class SmallFullyConnectedLayerDataset final : public FullyConnectedLayerDataset
 133 {
 134 public:
 135     SmallFullyConnectedLayerDataset()
 136     {
 137         // Conv -> FC
 138         add_config(TensorShape(8U, 1U, 1U), TensorShape(8U, 16U), TensorShape(16U), TensorShape(16U));
 139         // Conv -> FC
 140         add_config(TensorShape(1U, 1U, 1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U));
 141         // Conv -> FC
 142         add_config(TensorShape(9U, 5U, 7U), TensorShape(315U, 271U), TensorShape(271U), TensorShape(271U));
 143         // Conv -> FC (batched)
 144         add_config(TensorShape(9U, 5U, 7U, 3U), TensorShape(315U, 271U), TensorShape(271U), TensorShape(271U, 3U));
 145         // FC -> FC
 146         add_config(TensorShape(1U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U));
 147         // FC -> FC (batched)
 148         add_config(TensorShape(1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U));
 149         // FC -> FC
 150         add_config(TensorShape(201U), TensorShape(201U, 529U), TensorShape(529U), TensorShape(529U));
 151         // FC -> FC (batched)
 152         add_config(TensorShape(201U, 3U), TensorShape(201U, 529U), TensorShape(529U), TensorShape(529U, 3U));
 153
 154         add_config(TensorShape(9U, 5U, 7U, 3U, 2U), TensorShape(315U, 271U), TensorShape(271U), TensorShape(271U, 3U, 2U));
 155     }
 156 };
 157
 158 class LargeFullyConnectedLayerDataset final : public FullyConnectedLayerDataset
 159 {
 160 public:
 161     LargeFullyConnectedLayerDataset()
 162     {
 163         add_config(TensorShape(9U, 5U, 257U), TensorShape(11565U, 2123U), TensorShape(2123U), TensorShape(2123U));
 164         add_config(TensorShape(9U, 5U, 257U, 2U), TensorShape(11565U, 2123U), TensorShape(2123U), TensorShape(2123U, 2U));
 165         add_config(TensorShape(3127U), TensorShape(3127U, 989U), TensorShape(989U), TensorShape(989U));
 166         add_config(TensorShape(3127U, 2U), TensorShape(3127U, 989U), TensorShape(989U), TensorShape(989U, 2U));
 167
 168         add_config(TensorShape(9U, 5U, 257U, 2U, 3U), TensorShape(11565U, 2123U), TensorShape(2123U), TensorShape(2123U, 2U, 3U));
 169     }
 170 };
 171
 172 } // namespace datasets
 173 } // namespace test
 174 } // namespace arm_compute
 175 #endif /* ARM_COMPUTE_TEST_FULLYCONNECTED_LAYER_DATASET */