diff options
-rw-r--r-- | CMakeLists.txt | 2 | ||||
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 75064528c0..41db8661d3 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -447,7 +447,7 @@ if (BUILD_ARMNN_TFLITE_DELEGATE) set(ARMNN_SUB_PROJECT ON) set(ARMNN_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}) add_subdirectory(delegate) - add_definitions(-DARMNN_TF_LITE_DELEGATE) + add_definitions(-DARMNN_TFLITE_DELEGATE) endif() if(BUILD_BARE_METAL) diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 153fe5bcc7..ad462b2460 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -1060,7 +1060,7 @@ int main(int argc, const char* argv[]) ProgramOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::TfliteInterpreter) { - #if defined(ARMNN_TF_LITE_DELEGATE) + #if defined(ARMNN_TFLITE_DELEGATE) return TfLiteDelegateMainImpl(ProgramOptions.m_ExNetParams, ProgramOptions.m_RuntimeOptions); #else ARMNN_LOG(fatal) << "Not built with Arm NN Tensorflow-Lite delegate support."; |