diff options
Diffstat (limited to 'tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp')
| -rw-r--r-- | tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp b/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp index 282d3e67bd..b186c9b966 100644 --- a/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp +++ b/tests/benchmark_new/NEON/SYSTEM/AlexNet.cpp @@ -25,6 +25,7 @@ #include "arm_compute/core/Types.h" #include "arm_compute/runtime/NEON/functions/NEActivationLayer.h" #include "arm_compute/runtime/NEON/functions/NEConvolutionLayer.h" +#include "arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h" #include "arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h" #include "arm_compute/runtime/NEON/functions/NENormalizationLayer.h" #include "arm_compute/runtime/NEON/functions/NEPoolingLayer.h" @@ -57,6 +58,7 @@ using NEAlexNetFixture = AlexNetFixture<ITensor, Accessor, NEActivationLayer, NEConvolutionLayer, + NEDirectConvolutionLayer, NEFullyConnectedLayer, NENormalizationLayer, NEPoolingLayer, |