aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/benchmark_new/CL/SYSTEM/AlexNet.cpp')
-rw-r--r--tests/benchmark_new/CL/SYSTEM/AlexNet.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp b/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp
index e6d91d9706..271ed9904e 100644
--- a/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp
+++ b/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp
@@ -28,6 +28,7 @@
#include "arm_compute/runtime/CL/CLTensorAllocator.h"
#include "arm_compute/runtime/CL/functions/CLActivationLayer.h"
#include "arm_compute/runtime/CL/functions/CLConvolutionLayer.h"
+#include "arm_compute/runtime/CL/functions/CLDirectConvolutionLayer.h"
#include "arm_compute/runtime/CL/functions/CLFullyConnectedLayer.h"
#include "arm_compute/runtime/CL/functions/CLNormalizationLayer.h"
#include "arm_compute/runtime/CL/functions/CLPoolingLayer.h"
@@ -48,6 +49,7 @@ using CLAlexNetFixture = AlexNetFixture<ICLTensor,
CLAccessor,
CLActivationLayer,
CLConvolutionLayer,
+ CLDirectConvolutionLayer,
CLFullyConnectedLayer,
CLNormalizationLayer,
CLPoolingLayer,