Merge pull request #201 from helq/master

Update to tensorflow 1.9

ChangeLog.md

@@ -1,5 +1,8 @@
 # ChangeLog
 
+## v0.2.0.1
+- Switch to tensorflow 1.9.
+
 ## v0.2.0.0
 - Switch to tensorflow 1.8.
 - Switch to proto-lens 0.2.2.

ci_build/Dockerfile

@@ -3,7 +3,7 @@
 # stack to be installed on the host.  This comes at the expense of
 # flexibility.
 
-FROM tensorflow/tensorflow:1.8.0
+FROM tensorflow/tensorflow:1.9.0
 LABEL maintainer="TensorFlow authors <[email protected]>"
 
 # The build context directory is the top of the tensorflow-haskell
@@ -32,8 +32,8 @@ RUN \
     curl -O -L https://github.com/google/protobuf/releases/download/v3.2.0/protoc-3.2.0-linux-x86_64.zip && \
     unzip -d /usr/local protoc-3.2.0-linux-x86_64.zip bin/protoc && \
     chmod 755 /usr/local/bin/protoc && \
-    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.8.0.tar.gz && \
-    tar zxf libtensorflow-cpu-linux-x86_64-1.8.0.tar.gz -C /usr/local && \
+    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.9.0.tar.gz && \
+    tar zxf libtensorflow-cpu-linux-x86_64-1.9.0.tar.gz -C /usr/local && \
     ldconfig && \
     stack setup && \
     stack test --only-dependencies

docker/Dockerfile

@@ -1,6 +1,6 @@
 # Prepare the image with:
 #   docker build -t tensorflow/haskell:v0 docker
-FROM tensorflow/tensorflow:1.8.0
+FROM tensorflow/tensorflow:1.9.0
 LABEL maintainer="TensorFlow authors <[email protected]>"
 
 RUN apt-get update
@@ -27,8 +27,8 @@ RUN \
     curl -O -L https://github.com/google/protobuf/releases/download/v3.2.0/protoc-3.2.0-linux-x86_64.zip && \
     unzip -d /usr/local protoc-3.2.0-linux-x86_64.zip bin/protoc && \
     chmod 755 /usr/local/bin/protoc && \
-    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.8.0.tar.gz && \
-    tar zxf libtensorflow-cpu-linux-x86_64-1.8.0.tar.gz -C /usr/local && \
+    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.9.0.tar.gz && \
+    tar zxf libtensorflow-cpu-linux-x86_64-1.9.0.tar.gz -C /usr/local && \
     ldconfig
 
 ENV LANG en_US.UTF-8

docker/gpu/Dockerfile

@@ -1,6 +1,6 @@
 # Prepare the image with:
-#   docker build -t tensorflow/haskell:1.8.0-gpu docker/gpu
-FROM gcr.io/tensorflow/tensorflow:1.8.0-gpu
+#   docker build -t tensorflow/haskell:1.9.0-gpu docker/gpu
+FROM gcr.io/tensorflow/tensorflow:1.9.0-gpu
 LABEL maintainer="TensorFlow authors <[email protected]>"
 
 RUN apt-get update
@@ -27,8 +27,8 @@ RUN \
     curl -O -L https://github.com/google/protobuf/releases/download/v3.2.0/protoc-3.2.0-linux-x86_64.zip && \
     unzip -d /usr/local protoc-3.2.0-linux-x86_64.zip bin/protoc && \
     chmod 755 /usr/local/bin/protoc && \
-    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-1.8.0.tar.gz && \
-    tar zxf libtensorflow-gpu-linux-x86_64-1.8.0.tar.gz -C /usr/local && \
+    curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-1.9.0.tar.gz && \
+    tar zxf libtensorflow-gpu-linux-x86_64-1.9.0.tar.gz -C /usr/local && \
     ldconfig
 
 ENV LANG en_US.UTF-8

tensorflow-core-ops/Setup.hs

@@ -94,20 +94,27 @@ blackList =
     , "FlatMapDataset"
     , "For"
     , "GeneratorDataset"
+    , "GroupByReducerDataset"
     , "GroupByWindowDataset"
     , "If"
     , "InterleaveDataset"
     , "MapAndBatchDataset"
+    , "MapAndBatchDatasetV2"
     , "MapDataset"
     , "MapDataset"
     , "OneShotIterator"
     , "ParallelInterleaveDataset"
     , "ParallelMapDataset"
+    , "PartitionedCall"
     , "RemoteCall"
     , "ScanDataset"
     , "SymbolicGradient"
     , "TPUReplicate"
     , "While"
+    , "XlaIf"
+    , "XlaLaunch"
+    , "XlaReduceWindow"
+    , "XlaWhile"
     , "_If"
     , "_While"
     ]

tools/install_macos_dependencies.sh

@@ -27,7 +27,7 @@ fi
 TMP_DIR=$(mktemp -d)
 
 echo "Downloading libtensorflow..."
-curl https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-1.8.0.tar.gz > "$TMP_DIR/libtensorflow.tar.gz"
+curl https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-1.9.0.tar.gz > "$TMP_DIR/libtensorflow.tar.gz"
 
 echo "Extracting and copying libtensorflow..."
 tar zxf "$TMP_DIR/libtensorflow.tar.gz" -C "$TMP_DIR"