diff options
Diffstat (limited to 'src/armnnSerializer/test')
-rw-r--r-- | src/armnnSerializer/test/ActivationSerializationTests.cpp | 7 | ||||
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 5 |
2 files changed, 9 insertions, 3 deletions
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> |