From 4a5470691e239a38adee53f78d0204f10d4ef587 Mon Sep 17 00:00:00 2001
From: handong <455808519@qq.com>
Date: Thu, 26 Sep 2019 13:08:39 +0800
Subject: [PATCH] fix .so file name error on linux and macos
---
.../java/src/main/java/org/tensorflow/NativeLibrary.java | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/tensorflow/java/src/main/java/org/tensorflow/NativeLibrary.java b/tensorflow/java/src/main/java/org/tensorflow/NativeLibrary.java
index 53ef224640d..1c1bdac4763 100644
--- a/tensorflow/java/src/main/java/org/tensorflow/NativeLibrary.java
+++ b/tensorflow/java/src/main/java/org/tensorflow/NativeLibrary.java
@@ -71,6 +71,12 @@ final class NativeLibrary {
log("frameworkResourceName: " + frameworkResourceName);
final InputStream frameworkResource =
NativeLibrary.class.getClassLoader().getResourceAsStream(frameworkResourceName);
+ if ("libtensorflow_framework.so".equals(frameworkLibName)) {
+ frameworkLibName = "libtensorflow_framework.so.1";
+ }
+ if ("libtensorflow_framework.dylib".equals(frameworkLibName)) {
+ frameworkLibName = "libtensorflow_framework.1.dylib";
+ }
// Do not complain if the framework resource wasn't found. This may just mean that we're
// building with --config=monolithic (in which case it's not needed and not included).
if (jniResource == null) {