aboutsummaryrefslogtreecommitdiff
path: root/delegate/CMakeLists.txt
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/CMakeLists.txt')
-rw-r--r--delegate/CMakeLists.txt39
1 files changed, 31 insertions, 8 deletions
diff --git a/delegate/CMakeLists.txt b/delegate/CMakeLists.txt
index be18052548..aba27dfdaa 100644
--- a/delegate/CMakeLists.txt
+++ b/delegate/CMakeLists.txt
@@ -15,7 +15,31 @@ list(APPEND armnnDelegate_sources
include/armnn_delegate.hpp
include/DelegateOptions.hpp
src/armnn_delegate.cpp
- src/DelegateOptions.cpp)
+ src/DelegateOptions.cpp
+ src/Activation.hpp
+ src/ArgMinMax.hpp
+ src/BatchSpace.hpp
+ src/Comparison.hpp
+ src/Convolution.hpp
+ src/Control.hpp
+ src/DelegateUtils.hpp
+ src/ElementwiseBinary.hpp
+ src/ElementwiseUnary.hpp
+ src/Fill.hpp
+ src/FullyConnected.hpp
+ src/Gather.hpp
+ src/Lstm.hpp
+ src/Normalization.hpp
+ src/Pad.hpp
+ src/Pooling.hpp
+ src/Quantization.hpp
+ src/Redefine.hpp
+ src/Resize.hpp
+ src/Round.hpp
+ src/Slice.hpp
+ src/Softmax.hpp
+ src/SpaceDepth.hpp
+ src/Transpose.hpp)
add_library(armnnDelegate SHARED ${armnnDelegate_sources})
@@ -52,10 +76,6 @@ target_include_directories(armnnDelegate
PRIVATE
${TfLite_INCLUDE_DIR})
-target_include_directories(armnnDelegate
- SYSTEM PRIVATE
- "${TfLite_Schema_INCLUDE_PATH}")
-
## Add Flatbuffers dependency
find_package(Flatbuffers REQUIRED MODULE)
@@ -68,7 +88,10 @@ target_include_directories(armnnDelegate
set(armnnDelegate_unittest_sources)
list(APPEND armnnDelegate_unittest_sources
- src/test/ArmnnDelegateTest.cpp)
+ src/test/AbsTest.cpp
+ src/test/ArmnnDelegateTest.cpp
+ src/test/ElementwiseUnaryTestHelper.hpp
+ src/test/SqrtTest.cpp)
add_executable(DelegateUnitTests ${armnnDelegate_unittest_sources})
target_include_directories(DelegateUnitTests PRIVATE src)
@@ -81,8 +104,8 @@ target_include_directories(DelegateUnitTests
${TfLite_INCLUDE_DIR})
target_include_directories(DelegateUnitTests
- SYSTEM PRIVATE
- "${TfLite_Schema_INCLUDE_PATH}")
+ PRIVATE
+ ${Flatbuffers_INCLUDE_DIR})
####################################################
## Export targets