Polishing.

Author: peci1

cmake/detect_tf_catkin.cmake

@@ -33,8 +33,8 @@ endif()
 set(ADDITIONAL_LIBS "")
 if(${HAS_TENSORFLOW_GPU})
   if (CMAKE_CXX_COMPILER_ID MATCHES "GNU" )
-    find_library(gomp_LIBRARY gomp PATHS ${tensorflow_catkin_LIBRARY_DIRS})
-    set(ADDITIONAL_LIBS ${gomp_LIBRARY})
+    find_library(libgomp1_LIBRARIES gomp PATHS ${tensorflow_catkin_LIBRARY_DIRS})
+    set(ADDITIONAL_LIBS ${libgomp1_LIBRARIES})
   endif()
 endif()