[mlgo] Use LLVM_HAVE_TFLITE instead of LLVM_HAVE_TF_API

This patch replaces uses of LLVM_HAVE_TF_API with LLVM_HAVE_TFLITE in a couple of CMakeLists.txt. Now that 842b0d0 has landed, we now have: LLVM_HAVE_TF_API is defined if and only if LLVM_HAVE_TFLITE evaluates to true in the CMake variable world (assuming that you do not set LLVM_HAVE_TF_API on the cmake invocation). FWIW, the story is a little different in the C++ macro world, where: LLVM_HAVE_TF_API is defined if and only if LLVM_HAVE_TFLITE is defined This is why edc83a1 consisted only of mechanical replacements. Differential Revision: https://reviews.llvm.org/D140061

Author: kazutakahirata

llvm/lib/Analysis/CMakeLists.txt

@@ -1,4 +1,4 @@
-if (DEFINED LLVM_HAVE_TF_AOT OR DEFINED LLVM_HAVE_TF_API)
+if (DEFINED LLVM_HAVE_TF_AOT OR LLVM_HAVE_TFLITE)
  include(TensorFlowCompile)
  set(LLVM_INLINER_MODEL_PATH_DEFAULT "models/inliner-Oz")
 
@@ -17,18 +17,14 @@ if (DEFINED LLVM_HAVE_TF_AOT OR DEFINED LLVM_HAVE_TF_API)
  )
  endif()
 
- if (DEFINED LLVM_HAVE_TF_API)
- if (LLVM_HAVE_TFLITE)
- build_proto(
- tensorflow/core/protobuf/error_codes
- tensorflow/core/example/feature
- tensorflow/core/example/example)
- list(APPEND MLLinkDeps 
- tensorflow-lite::tensorflow-lite
- protobuf::libprotobuf)
- else()
- list(APPEND MLLinkDeps ${tensorflow_c_api} ${tensorflow_fx})
- endif()
+ if (LLVM_HAVE_TFLITE)
+ build_proto(
+ tensorflow/core/protobuf/error_codes
+ tensorflow/core/example/feature
+ tensorflow/core/example/example)
+ list(APPEND MLLinkDeps
+ tensorflow-lite::tensorflow-lite
+ protobuf::libprotobuf)
  endif()
 endif()
 

llvm/unittests/Analysis/CMakeLists.txt

@@ -7,7 +7,7 @@ set(LLVM_LINK_COMPONENTS
  )
 
 set(MLGO_TESTS TFUtilsTest.cpp TrainingLoggerTest.cpp)
-if (DEFINED LLVM_HAVE_TF_API)
+if (LLVM_HAVE_TFLITE)
  LIST(APPEND EXTRA_TESTS ${MLGO_TESTS})
 else()
  LIST(APPEND LLVM_OPTIONAL_SOURCES ${MLGO_TESTS})