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
 {