aboutsummaryrefslogtreecommitdiff
path: root/tests/model_objects
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_objects')
-rw-r--r--tests/model_objects/AlexNet.h2
-rw-r--r--tests/model_objects/LeNet5.h2
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/model_objects/AlexNet.h b/tests/model_objects/AlexNet.h
index 672942d365..dad68324c1 100644
--- a/tests/model_objects/AlexNet.h
+++ b/tests/model_objects/AlexNet.h
@@ -24,8 +24,8 @@
#ifndef __ARM_COMPUTE_TEST_MODEL_OBJECTS_ALEXNET_H__
#define __ARM_COMPUTE_TEST_MODEL_OBJECTS_ALEXNET_H__
+#include "tests/AssetsLibrary.h"
#include "tests/Globals.h"
-#include "tests/TensorLibrary.h"
#include "tests/Utils.h"
#include <memory>
diff --git a/tests/model_objects/LeNet5.h b/tests/model_objects/LeNet5.h
index 3a9046c205..d3e72b0010 100644
--- a/tests/model_objects/LeNet5.h
+++ b/tests/model_objects/LeNet5.h
@@ -24,8 +24,8 @@
#ifndef __ARM_COMPUTE_TEST_MODEL_OBJECTS_LENET5_H__
#define __ARM_COMPUTE_TEST_MODEL_OBJECTS_LENET5_H__
+#include "tests/AssetsLibrary.h"
#include "tests/Globals.h"
-#include "tests/TensorLibrary.h"
#include "tests/Utils.h"
#include <memory>