diff options
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 9 | ||||
-rw-r--r-- | src/armnnSerializer/SerializerUtils.hpp | 3 | ||||
-rw-r--r-- | src/armnnSerializer/test/ActivationSerializationTests.cpp | 7 | ||||
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 5 |
4 files changed, 15 insertions, 9 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; diff --git a/src/armnnSerializer/SerializerUtils.hpp b/src/armnnSerializer/SerializerUtils.hpp index 16b83e3ed7..48553334ba 100644 --- a/src/armnnSerializer/SerializerUtils.hpp +++ b/src/armnnSerializer/SerializerUtils.hpp @@ -4,8 +4,7 @@ // #pragma once -#include <armnn/ArmNN.hpp> - +#include <armnn/Types.hpp> #include <ArmnnSchema_generated.h> namespace armnnSerializer diff --git a/src/armnnSerializer/test/ActivationSerializationTests.cpp b/src/armnnSerializer/test/ActivationSerializationTests.cpp index 34e99f66db..0362412d0c 100644 --- a/src/armnnSerializer/test/ActivationSerializationTests.cpp +++ b/src/armnnSerializer/test/ActivationSerializationTests.cpp @@ -3,10 +3,13 @@ // SPDX-License-Identifier: MIT // -#include <armnnDeserializer/IDeserializer.hpp> -#include <armnn/ArmNN.hpp> +#include <armnn/Descriptors.hpp> #include <armnn/INetwork.hpp> +#include <armnn/IRuntime.hpp> +#include <armnnDeserializer/IDeserializer.hpp> + #include "../Serializer.hpp" + #include <sstream> #include <boost/core/ignore_unused.hpp> diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 3e67cf062f..2b13109725 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -5,8 +5,11 @@ #include "../Serializer.hpp" -#include <armnn/ArmNN.hpp> +#include <armnn/Descriptors.hpp> #include <armnn/INetwork.hpp> +#include <armnn/TypesUtils.hpp> +#include <armnn/LstmParams.hpp> +#include <armnn/QuantizedLstmParams.hpp> #include <armnnDeserializer/IDeserializer.hpp> #include <random> |