diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2019-02-11 17:01:27 +0000 |
|---|---|---|
| committer | Mike Kelly <mike.kelly@arm.com> | 2019-02-11 17:01:27 +0000 |
| commit | 8c1701a2d9c1da0e1decb2afdc2093aa88810242 (patch) | |
| tree | 870ee9af506bb468c513214ab539f41aeb1e34dc /CMakeLists.txt | |
| parent | a40521a70e73d20a060fa2df0e83b02c4f1c6139 (diff) | |
| download | armnn-8c1701a2d9c1da0e1decb2afdc2093aa88810242.tar.gz | |
IVGCVSW-2531 Serialize a simple ArmNN Network
Change-Id: I68cf5072aca6e3a8b3b8c57e19b6d417cd5813fc
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index ea0f2c31ae..71b4dfe8b9 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -158,8 +158,10 @@ if(BUILD_ARMNN_SERIALIZER) set(armnn_serializer_sources) list(APPEND armnn_serializer_sources - src/armnnSerializer/Schema_generated.h include/armnnDeserializeParser/IDeserializeParser.hpp + src/armnnSerializer/Schema_generated.h + src/armnnSerializer/Serializer.hpp + src/armnnSerializer/Serializer.cpp src/armnnDeserializeParser/DeserializeParser.hpp src/armnnDeserializeParser/DeserializeParser.cpp ) @@ -188,6 +190,7 @@ list(APPEND armnn_sources include/armnn/IProfiler.hpp include/armnn/IRuntime.hpp include/armnn/LayerSupport.hpp + include/armnn/LayerVisitorBase.hpp include/armnn/LstmParams.hpp include/armnn/NetworkFwd.hpp include/armnn/Optional.hpp @@ -303,7 +306,6 @@ list(APPEND armnn_sources src/armnn/LayersFwd.hpp src/armnn/LayerSupportCommon.hpp src/armnn/LayerSupport.cpp - src/armnn/LayerVisitorBase.hpp src/armnn/LoadedNetwork.cpp src/armnn/LoadedNetwork.hpp src/armnn/Network.cpp @@ -546,6 +548,7 @@ if(BUILD_UNIT_TESTS) enable_language(ASM) list(APPEND unittest_sources src/armnnSerializer/Schema_generated.h + src/armnnSerializer/test/SerializerTests.cpp src/armnnDeserializeParser/test/DeserializeAdd.cpp src/armnnDeserializeParser/test/ParserFlatbuffersSerializeFixture.hpp src/armnnDeserializeParser/test/SchemaSerialize.s @@ -581,10 +584,6 @@ if(BUILD_UNIT_TESTS) target_include_directories(UnitTests PRIVATE src/armnnUtils) target_include_directories(UnitTests PRIVATE src/backends) - if(BUILD_ARMNN_SERIALIZER) - target_include_directories(UnitTests SYSTEM PRIVATE "${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer") - endif() - if(BUILD_TF_LITE_PARSER) target_include_directories(UnitTests SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}") endif() @@ -613,6 +612,7 @@ if(BUILD_UNIT_TESTS) endif() if(BUILD_ARMNN_SERIALIZER) + target_include_directories(UnitTests SYSTEM PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src/armnnSerializer) target_link_libraries(UnitTests armnnSerializer) endif() |