From 232cfc2a544c874b35cc2a949e9fa71f41690565 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 13 Mar 2019 18:33:10 +0000 Subject: 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 --- CMakeLists.txt | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) (limited to 'CMakeLists.txt') 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) -- cgit v1.2.1