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