[gpu]: Update to 1.4.0

Author: asimshankar

gpu/README.md

@@ -2,7 +2,7 @@
 
 The TensorFlow Java API is distributed in the
 [org.tensorflow:tensorflow](http://mvnrepository.com/artifact/org.tensorflow/tensorflow)
-maven package. As of TensorFlow 1.2.1, this package did not include GPU support
+maven package. As of TensorFlow 1.4.0, this package did not include GPU support
 by default. However, by making the GPU-enabled native libraries available
 to the JVM, GPUs can be utilized. This example demonstrates that.
 
@@ -26,7 +26,7 @@ to the JVM, GPUs can be utilized. This example demonstrates that.
     ```
     output: (MatMul): /job:localhost/replica:0/task:0/cpu:0
     input: (Placeholder): /job:localhost/replica:0/task:0/cpu:0
-    TensorFlow version: 1.2.1
+    TensorFlow version: 1.4.0
 
     Input : [[1.0, 2.0], [3.0, 4.0]]
     Output: [[7.0, 10.0], [15.0, 22.0]]
@@ -43,8 +43,8 @@ CUDA-enabled Java native libraries need to be made available to the JVM.
 1.  Download and extract the CUDA-enabled TensorFlow native library:
 
     ```
-    curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-1.2.1.tar.gz" |
-      tar -xz ./libtensorflow_jni.so
+    curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-1.4.0.tar.gz" |
+      tar -xz
     ```
 
     (For more detailed instructions see:
@@ -65,7 +65,7 @@ JVM will result in something like this on the console:
 ```
 output: (MatMul): /job:localhost/replica:0/task:0/gpu:0
 input: (Placeholder): /job:localhost/replica:0/task:0/gpu:0
-TensorFlow version: 1.2.1
+TensorFlow version: 1.4.0
 
 Input : [[1.0, 2.0], [3.0, 4.0]]
 Output: [[7.0, 10.0], [15.0, 22.0]]

gpu/pom.xml

@@ -14,12 +14,12 @@
       <dependency>
         <groupId>org.tensorflow</groupId>
         <artifactId>tensorflow</artifactId>
-        <version>1.2.1</version>
+        <version>1.4.0</version>
       </dependency>
       <dependency>
         <groupId>org.tensorflow</groupId>
         <artifactId>proto</artifactId>
-        <version>1.2.1</version>
+        <version>1.4.0</version>
       </dependency>
     </dependencies>
 </project>

gpu/src/main/java/HelloTF.java

@@ -5,6 +5,7 @@
 import org.tensorflow.Session;
 import org.tensorflow.Tensor;
 import org.tensorflow.TensorFlow;
+import org.tensorflow.Tensors;
 import org.tensorflow.framework.ConfigProto;
 
 public class HelloTF {
@@ -18,8 +19,9 @@ public static void main(String[] args) throws Exception {
       // Create a config that will dump out device placement of operations.
       ConfigProto config = ConfigProto.newBuilder().setLogDevicePlacement(true).build();
       try (Session sess = new Session(graph, config.toByteArray())) {
-        try (Tensor in = Tensor.create(new float[][] {{1, 2}, {3, 4}});
-            Tensor out = sess.runner().feed("input", in).fetch("output").run().get(0)) {
+        try (Tensor<Float> in = Tensors.create(new float[][] {{1, 2}, {3, 4}});
+            Tensor<Float> out =
+                sess.runner().feed("input", in).fetch("output").run().get(0).expect(Float.class)) {
           System.out.println("TensorFlow version: " + TensorFlow.version());
           System.out.println();
           print2x2Matrix("Input ", in);
@@ -29,7 +31,7 @@ public static void main(String[] args) throws Exception {
     }
   }
 
-  public static void print2x2Matrix(String tag, Tensor t) {
+  public static void print2x2Matrix(String tag, Tensor<Float> t) {
     float[][] m = new float[2][2];
     System.out.print(tag + ": ");
     System.out.println(Arrays.deepToString(t.copyTo(m)));