diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2019-03-13 18:33:10 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-03-13 18:33:10 +0000 |
| commit | 232cfc2a544c874b35cc2a949e9fa71f41690565 (patch) | |
| tree | a20bb3dfdc36f8d2fccac72136fed91b4ad9993c /CMakeLists.txt | |
| parent | db7c3912ff3b37dd1fdef88e957b302c72042cce (diff) | |
| download | armnn-232cfc2a544c874b35cc2a949e9fa71f41690565.tar.gz | |
IVGCVSW-2574 Create a converter application for all the supported ArmNN Networks
* Added Caffe, TfLite and Onnx support to ArmnnConverter
Change-Id: I9807064152f23a24d280118dbed0396eb7d47420
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 5aed9837a1..99f00d8e9c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -641,7 +641,7 @@ if(BUILD_UNIT_TESTS) addDllCopyCommands(UnitTests) endif() -if (BUILD_ARMNN_SERIALIZER AND BUILD_TF_PARSER) +if (BUILD_ARMNN_SERIALIZER AND (BUILD_TF_PARSER OR BUILD_TF_LITE_PARSER OR BUILD_ONNX_PARSER OR BUILD_CAFFE_PARSER)) set(ArmnnConverter_sources src/armnnConverter/ArmnnConverter.cpp) @@ -649,7 +649,22 @@ if (BUILD_ARMNN_SERIALIZER AND BUILD_TF_PARSER) target_include_directories(ArmnnConverter PRIVATE src/armnn) target_include_directories(ArmnnConverter PRIVATE src/armnnUtils) - target_link_libraries(ArmnnConverter armnnTfParser) + if(BUILD_CAFFE_PARSER) + target_link_libraries(ArmnnConverter armnnCaffeParser) + endif() + + if(BUILD_ONNX_PARSER) + target_link_libraries(ArmnnConverter armnnOnnxParser) + endif() + + if(BUILD_TF_PARSER) + target_link_libraries(ArmnnConverter armnnTfParser) + endif() + + if(BUILD_TF_LITE_PARSER) + target_link_libraries(ArmnnConverter armnnTfLiteParser) + endif() + target_link_libraries(ArmnnConverter armnnSerializer) target_link_libraries(ArmnnConverter armnn) |