tf-head: Fix undefined symbols by linking libtensorflow_cc.so.2 (horo…

…vod#3755)

Signed-off-by: Max H. Gerlach <git@maxgerlach.de>

cmake/Modules/FindTensorflow.cmake

@@ -25,6 +25,11 @@ if (LEN EQUAL "4")
     if (Tensorflow_VERSION VERSION_GREATER_EQUAL "2.6")
         # XLA implementations and helpers for resource variables are in _pywrap_tensorflow_internal.so
         set(Tensorflow_LIBRARIES "${Tensorflow_LIBRARIES} ${Tensorflow_LIB_PATH}/python/_pywrap_tensorflow_internal.so")
+        set(lib_tensorflow_cc_so_path "${Tensorflow_LIB_PATH}/libtensorflow_cc.so.2")
+        if (EXISTS "${lib_tensorflow_cc_so_path}")
+            # TF 2.12: https://github.com/tensorflow/tensorflow/pull/55941 moves implementations to new libtensorflow_cc.so
+            set(Tensorflow_LIBRARIES "${Tensorflow_LIBRARIES} ${lib_tensorflow_cc_so_path}")
+        endif()
     endif()
     message("Tensorflow_LIBRARIES := ${Tensorflow_LIBRARIES}")
     list(GET Tensorflow_OUTPUT 3 Tensorflow_COMPILE_FLAGS)