diff options
Diffstat (limited to 'tests/datasets/ReshapeLayerDataset.h')
| -rw-r--r-- | tests/datasets/ReshapeLayerDataset.h | 2 |
1 files changed, 2 insertions, 0 deletions
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 { |