diff options
Diffstat (limited to 'tests/validation/Datasets.h')
-rw-r--r-- | tests/validation/Datasets.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/validation/Datasets.h b/tests/validation/Datasets.h index ae76fb6be3..ab21787f45 100644 --- a/tests/validation/Datasets.h +++ b/tests/validation/Datasets.h @@ -78,12 +78,24 @@ struct is_dataset<arm_compute::test::Small1DShape> : boost::mpl::true_ /// Register the data set with Boost template <> +struct is_dataset<arm_compute::test::Small2DShapes> : boost::mpl::true_ +{ +}; + +/// Register the data set with Boost +template <> struct is_dataset<arm_compute::test::LargeShapes> : boost::mpl::true_ { }; /// Register the data set with Boost template <> +struct is_dataset<arm_compute::test::Large2DShapes> : boost::mpl::true_ +{ +}; + +/// Register the data set with Boost +template <> struct is_dataset<arm_compute::test::AllDataTypes> : boost::mpl::true_ { }; |