diff options
Diffstat (limited to 'tests/datasets/ScatterDataset.h')
-rw-r--r-- | tests/datasets/ScatterDataset.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/datasets/ScatterDataset.h b/tests/datasets/ScatterDataset.h index 09f6338432..d204d17855 100644 --- a/tests/datasets/ScatterDataset.h +++ b/tests/datasets/ScatterDataset.h @@ -113,12 +113,13 @@ private: std::vector<TensorShape> _dst_shapes{}; }; -class SmallScatterDataset final : public ScatterDataset +class Small1DScatterDataset final : public ScatterDataset { public: - SmallScatterDataset() + Small1DScatterDataset() { add_config(TensorShape(6U), TensorShape(6U), TensorShape(6U), TensorShape(6U)); + add_config(TensorShape(10U), TensorShape(2U), TensorShape(2U), TensorShape(10U)); } }; } // namespace datasets |