diff options
Diffstat (limited to 'tests/benchmark_new/NEON/SYSTEM')
-rw-r--r-- | tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp | 2 | ||||
-rw-r--r-- | tests/benchmark_new/NEON/SYSTEM/LeNet5.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp b/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp index 2a50bc9d85..8ed70f7a6e 100644 --- a/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp +++ b/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp @@ -57,7 +57,7 @@ using NEAlexNetFixture = AlexNetFixture<ITensor, TEST_SUITE(SYSTEM_TEST) TEST_SUITE(NEON) -REGISTER_FIXTURE_DATA_TEST_CASE(AlexNet, NEAlexNetFixture, +REGISTER_FIXTURE_DATA_TEST_CASE(AlexNet, NEAlexNetFixture, framework::DatasetMode::ALL, framework::dataset::combine(framework::dataset::make("Data type", { DataType::F32, DataType::QS8 }), framework::dataset::make("Batches", { 1, 4, 8 }))); diff --git a/tests/benchmark_new/NEON/SYSTEM/LeNet5.cpp b/tests/benchmark_new/NEON/SYSTEM/LeNet5.cpp index 3957ce76b9..9a5c49e975 100644 --- a/tests/benchmark_new/NEON/SYSTEM/LeNet5.cpp +++ b/tests/benchmark_new/NEON/SYSTEM/LeNet5.cpp @@ -51,7 +51,7 @@ using NELeNet5Fixture = LeNet5Fixture<Tensor, TEST_SUITE(SYSTEM_TEST) TEST_SUITE(NEON) -REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5, NELeNet5Fixture, +REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5, NELeNet5Fixture, framework::DatasetMode::ALL, framework::dataset::make("Batches", { 1, 4, 8 })); TEST_SUITE_END() |