diff options
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 9 |
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; |