diff options
Diffstat (limited to 'tests/datasets')
-rw-r--r-- | tests/datasets/PoolingTypesDataset.h | 2 | ||||
-rw-r--r-- | tests/datasets/ReshapeLayerDataset.h | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/tests/datasets/PoolingTypesDataset.h b/tests/datasets/PoolingTypesDataset.h index 5994d6ecd0..fbe2fba24c 100644 --- a/tests/datasets/PoolingTypesDataset.h +++ b/tests/datasets/PoolingTypesDataset.h @@ -35,6 +35,7 @@ namespace test { namespace datasets { +/** [PoolingTypes datasets] **/ class PoolingTypes final : public framework::dataset::ContainerDataset<std::vector<PoolingType>> { public: @@ -46,6 +47,7 @@ public: { } }; +/** [PoolingTypes datasets] **/ } // namespace datasets } // namespace test } // namespace arm_compute diff --git a/tests/datasets/ReshapeLayerDataset.h b/tests/datasets/ReshapeLayerDataset.h index 6af65a1c20..1a8932fb85 100644 --- a/tests/datasets/ReshapeLayerDataset.h +++ b/tests/datasets/ReshapeLayerDataset.h @@ -34,6 +34,7 @@ namespace test { namespace datasets { +/** [ReshapeLayer datasets] **/ class ReshapeLayerDataset { public: @@ -96,6 +97,7 @@ private: std::vector<TensorShape> _in_shapes{}; std::vector<TensorShape> _out_shapes{}; }; +/** [ReshapeLayer datasets] **/ class SmallReshapeLayerDataset final : public ReshapeLayerDataset { |