diff options
| author | Måns Nilsson <mans.nilsson@arm.com> | 2020-09-14 16:30:56 +0200 |
|---|---|---|
| committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2020-09-15 07:45:27 +0000 |
| commit | 98e379c83dd24619752e72e7aefdc15484813652 (patch) | |
| tree | 68a5b5058dd68962754c3c19de969b3673b9a498 | |
| parent | 101ccefb56ab032a7fc0ae4fe0934a906363edb0 (diff) | |
| download | ethos-u-core-software-98e379c83dd24619752e72e7aefdc15484813652.tar.gz | |
Add back TAGS=cmsis-nn in tflu_gen target
In case a CMSIS-NN operator is not supported by Ethos-U performance
would otherwise be degraded.
Change-Id: I0b688403b0a7c33cde1343f4e64ba14e901a19d7
| -rw-r--r-- | tensorflow.cmake | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow.cmake b/tensorflow.cmake index 66a63d4..27a8918 100644 --- a/tensorflow.cmake +++ b/tensorflow.cmake @@ -46,6 +46,8 @@ set(TFLU_OPTIMIZATION_LEVEL CACHE STRING "Tensorflow Lite Micro optimization lev if(CORE_SOFTWARE_ACCELERATOR STREQUAL NPU) set(TFLU_ETHOSU_LIBS $<TARGET_FILE:ethosu_core_driver>) + # Set preference for ethos-u over cmsis-nn + list(APPEND TFLU_TAGS "cmsis-nn") list(APPEND TFLU_TAGS "ethos-u") elseif(CORE_SOFTWARE_ACCELERATOR STREQUAL CMSIS-NN) list(APPEND TFLU_TAGS "cmsis-nn") |