aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp9
1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index be6fa64b0a..91b62413d8 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -5,17 +5,18 @@
#include "Serializer.hpp"
-#include "SerializerUtils.hpp"
-
-#include <armnn/ArmNN.hpp>
+#include <armnn/Descriptors.hpp>
+#include <armnn/LstmParams.hpp>
+#include <armnn/QuantizedLstmParams.hpp>
#include <iostream>
#include <boost/core/ignore_unused.hpp>
#include <boost/numeric/conversion/cast.hpp>
-
#include <flatbuffers/util.h>
+#include "SerializerUtils.hpp"
+
using namespace armnn;
namespace fb = flatbuffers;
namespace serializer = armnnSerializer;