diff options
Diffstat (limited to 'tests/benchmark_new/NEON/DirectConvolutionLayer.cpp')
| -rw-r--r-- | tests/benchmark_new/NEON/DirectConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/benchmark_new/NEON/DirectConvolutionLayer.cpp b/tests/benchmark_new/NEON/DirectConvolutionLayer.cpp index a63cbae8e5..6610fd3ec9 100644 --- a/tests/benchmark_new/NEON/DirectConvolutionLayer.cpp +++ b/tests/benchmark_new/NEON/DirectConvolutionLayer.cpp @@ -41,7 +41,7 @@ using NEDirectConvolutionLayerFixture = ConvolutionLayerFixture<Tensor, NEDirect TEST_SUITE(NEON) -REGISTER_FIXTURE_DATA_TEST_CASE(DirectConvolutionLayer, NEDirectConvolutionLayerFixture, +REGISTER_FIXTURE_DATA_TEST_CASE(DirectConvolutionLayer, NEDirectConvolutionLayerFixture, framework::DatasetMode::ALL, framework::dataset::combine(framework::dataset::combine(datasets::DirectConvolutionLayerDataset(), framework::dataset::make("Data type", { DataType::F32, DataType::QS8 })), framework::dataset::make("Batches", { 1, 4, 8 }))); |