diff options
Diffstat (limited to 'tests/validation/Datasets.h')
-rw-r--r-- | tests/validation/Datasets.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/validation/Datasets.h b/tests/validation/Datasets.h index 33776d2e44..e33cc62a5e 100644 --- a/tests/validation/Datasets.h +++ b/tests/validation/Datasets.h @@ -34,6 +34,8 @@ #include "dataset/GEMMDataset.h" #include "dataset/ImageDatasets.h" #include "dataset/InterpolationPolicyDataset.h" +#include "dataset/MatrixPatternDataset.h" +#include "dataset/NonLinearFilterFunctionDataset.h" #include "dataset/NormalizationTypeDataset.h" #include "dataset/PoolingLayerDataset.h" #include "dataset/PoolingTypesDataset.h" @@ -250,6 +252,18 @@ template <> struct is_dataset<arm_compute::test::ThresholdDataset> : boost::mpl::true_ { }; + +/// Register the data set with Boost +template <> +struct is_dataset<arm_compute::test::NonLinearFilterFunctions> : boost::mpl::true_ +{ +}; + +/// Register the data set with Boost +template <> +struct is_dataset<arm_compute::test::MatrixPatterns> : boost::mpl::true_ +{ +}; } } } |