diff --git a/tensorflow-core/tensorflow-core-native/scripts/dist_download.sh b/tensorflow-core/tensorflow-core-native/scripts/dist_download.sh index 39235d137be..20068238fc3 100755 --- a/tensorflow-core/tensorflow-core-native/scripts/dist_download.sh +++ b/tensorflow-core/tensorflow-core-native/scripts/dist_download.sh @@ -37,6 +37,9 @@ if [[ -n "$WHEEL_URL" ]]; then curl -L $WHEEL_URL --output 'tensorflow.whl' fi yes | unzip -q -u 'tensorflow.whl' # use 'yes' because for some reasons -u does not work on Windows + if [[ "$PLATFORM" == "linux-arm64" ]]; then + cp $DOWNLOAD_FOLDER/tensorflow.libs/* $DOWNLOAD_FOLDER/tensorflow/ + fi fi if [[ -n "$CLIB_URL" ]]; then @@ -51,6 +54,9 @@ cd tensorflow if [[ "$PLATFORM" =~ "linux" ]]; then ln -fs libtensorflow_cc.so.2 libtensorflow_cc.so ln -fs libtensorflow_framework.so.2 libtensorflow_framework.so + if [[ "$PLATFORM" == "linux-arm64" ]]; then + ln -fs libomp-*.so.5 libomp.so + fi elif [[ "$PLATFORM" =~ "macosx" ]]; then ln -fs libtensorflow_cc.2.dylib libtensorflow_cc.dylib ln -fs libtensorflow_framework.2.dylib libtensorflow_framework.dylib diff --git a/tensorflow-core/tensorflow-core-native/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java b/tensorflow-core/tensorflow-core-native/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java index ff9f02709e3..eefd6893597 100644 --- a/tensorflow-core/tensorflow-core-native/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java +++ b/tensorflow-core/tensorflow-core-native/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java @@ -62,6 +62,9 @@ }, link = {"tensorflow_cc@.2", "tensorflow_framework@.2"}, resource = {"LICENSE", "THIRD_PARTY_TF_JNI_LICENSES"}), + @Platform( + value = {"linux-arm64"}, + link = {"tensorflow_cc@.2", "tensorflow_framework@.2", "omp@.5"}), @Platform( value = "windows", preload = {