[tensorflow] Upgrades tensorflow engine to 2.10.1 (deepjavalibrary#2440)

frankfliu · Mar 5, 2023 · 0ceaa1f · 0ceaa1f
1 parent 79151ad
commit 0ceaa1f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/engines/tensorflow/tensorflow-api/build.gradle b/engines/tensorflow/tensorflow-api/build.gradle
@@ -7,7 +7,7 @@ configurations {
 dependencies {
     api "org.bytedeco:javacpp:${javacpp_version}"
     api "com.google.protobuf:protobuf-java:${protobuf_version}"
-    api "org.tensorflow:tensorflow-core-api:0.4.2"
+    api "org.tensorflow:tensorflow-core-api:0.5.0"
 
     exclusion "org.bytedeco:javacpp:${javacpp_version}"
     exclusion "com.google.protobuf:protobuf-java:${protobuf_version}"
@@ -19,6 +19,7 @@ jar {
             it.isDirectory() ? it : zipTree(it)
         }
     }
+    exclude "module-info.class"
 }
 
 publishing {

diff --git a/engines/tensorflow/tensorflow-native/build.gradle b/engines/tensorflow/tensorflow-native/build.gradle
@@ -13,11 +13,11 @@ configurations {
 }
 
 dependencies {
-    download "org.tensorflow:tensorflow-core-api:0.4.2:linux-x86_64-gpu"
-    download "org.tensorflow:tensorflow-core-api:0.4.2:linux-x86_64"
-    download "org.tensorflow:tensorflow-core-api:0.4.2:macosx-x86_64"
-    download "org.tensorflow:tensorflow-core-api:0.4.2:windows-x86_64"
-    download "org.tensorflow:tensorflow-core-api:0.4.2:windows-x86_64-gpu"
+    download "org.tensorflow:tensorflow-core-api:0.5.0:linux-x86_64-gpu"
+    download "org.tensorflow:tensorflow-core-api:0.5.0:linux-x86_64"
+    download "org.tensorflow:tensorflow-core-api:0.5.0:macosx-x86_64"
+    download "org.tensorflow:tensorflow-core-api:0.5.0:windows-x86_64"
+    download "org.tensorflow:tensorflow-core-api:0.5.0:windows-x86_64-gpu"
 }
 
 task uploadTensorflowNativeLibs() {
@@ -76,6 +76,7 @@ task uploadTensorflowNativeLibs() {
         delete("${buildDir}/native/org/")
         delete("${buildDir}/native/com/")
         delete("${buildDir}/native/google/")
+        delete("${buildDir}/native/module-info.class.gz")
 
         exec {
             commandLine "aws", "s3", "sync", "${buildDir}/native/", "s3://djl-ai/publish/tensorflow-${tensorflow_version}/"

diff --git a/gradle.properties b/gradle.properties
@@ -14,7 +14,7 @@ systemProp.org.gradle.internal.publish.checksums.insecure=true
 djl_version=0.22.0
 mxnet_version=1.9.1
 pytorch_version=1.13.1
-tensorflow_version=2.7.4
+tensorflow_version=2.10.1
 tflite_version=2.6.2
 trt_version=8.4.1
 onnxruntime_version=1.14.0
@@ -36,7 +36,7 @@ slf4j_version=1.7.36
 log4j_slf4j_version=2.19.0
 awssdk_version=2.19.32
 hadoop_version=3.3.4
-javacpp_version=1.5.6
+javacpp_version=1.5.8
 protobuf_version=3.21.9
 tablesaw_version=0.43.1
 spark_version=3.2.2