diff --git a/src/TensorFlow/OpGen.hs b/src/TensorFlow/OpGen.hs
--- a/src/TensorFlow/OpGen.hs
+++ b/src/TensorFlow/OpGen.hs
@@ -150,7 +150,7 @@
     , "import Data.Complex (Complex)"
     , "import Data.Int (Int8, Int16, Int32, Int64)"
     , "import Data.Proxy (Proxy(Proxy))"
-    , "import Data.Word (Word8, Word16)"
+    , "import Data.Word (Word8, Word16, Word32, Word64)"
     , "import Lens.Family2 ((.~), (&))"
     , "import TensorFlow.Build"
     , "import TensorFlow.BuildOp"
@@ -415,9 +415,12 @@
 dtTypeToHaskell DT_QUINT8 = "Data.Word.Word8"  -- TODO(gnezdo): make unique
 dtTypeToHaskell DT_STRING = "Data.ByteString.ByteString"
 dtTypeToHaskell DT_UINT16 = "Data.Word.Word16"
+dtTypeToHaskell DT_UINT32 = "Data.Word.Word32"
+dtTypeToHaskell DT_UINT64 = "Data.Word.Word64"
 dtTypeToHaskell DT_HALF = "Data.Word.Word16"  -- TODO(gnezdo): make unique
 dtTypeToHaskell DT_UINT8 = "Data.Word.Word8"
 dtTypeToHaskell DT_RESOURCE = "ResourceHandle"
+dtTypeToHaskell DT_VARIANT = "Variant"
 dtTypeToHaskell x =
     Text.pack $ "Unsupported type in dtTypeToHaskell: " ++ show x
 
diff --git a/tensorflow-opgen.cabal b/tensorflow-opgen.cabal
--- a/tensorflow-opgen.cabal
+++ b/tensorflow-opgen.cabal
@@ -1,5 +1,5 @@
 name:                tensorflow-opgen
-version:             0.1.0.0
+version:             0.2.0.0
 synopsis:            Code generation for TensorFlow operations.
 description:         Please see README.md
 homepage:            https://github.com/tensorflow/haskell#readme
@@ -17,7 +17,7 @@
   exposed-modules:     TensorFlow.OpGen.ParsedOp
                      , TensorFlow.OpGen
   build-depends:  proto-lens == 0.2.*
-                , tensorflow-proto == 0.1.*
+                , tensorflow-proto == 0.2.*
                 , base >= 4.7 && < 5
                 , bytestring
                 , containers