experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix .so file name error on linux and macos

Browse Source
This commit is contained in:
handong 2019-09-26 13:08:39 +08:00
parent 1b8f5bc801
commit 4a5470691e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/java/src/main/java/org/tensorflow/NativeLibrary.java
View File

@ -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) {