Merge pull request #465 from JuliaComputing/kf/typemapping

Round out TensorFlow type mapping

Author: malmaud

src/constants.jl

@@ -17,8 +17,11 @@
   TF_QUINT16 = 16,
   TF_UINT16 = 17,
   TF_COMPLEX128 = 18,
-  TF_HALF = 19)
-
+  TF_HALF = 19,
+  TF_RESOURCE = 20,
+  TF_VARIANT = 21,
+  TF_UINT32 = 22,
+  TF_UINT64 = 23)
 
   @enum(TF_Code,
     TF_OK = 0,

src/core.jl

@@ -1372,10 +1372,13 @@ function Base.eltype(t::RawTensor)
     tf_to_jl_type(tf_type)
 end
 
-const type_map = Dict(TF_UINT8=>UInt8, TF_FLOAT=>Float32, TF_INT32=>Int32,
-                      TF_INT64=>Int64, TF_DOUBLE=>Float64, TF_STRING=>String,
-                      TF_BOOL=>Bool, TF_COMPLEX64=>ComplexF32,
-                      TF_COMPLEX128=>ComplexF64)
+const type_map = Dict(TF_INT8=>Int8, TF_UINT8=>UInt8,
+                      TF_INT16=>Int16, TF_UINT16=>UInt16,
+                      TF_INT32=>Int32, TF_UINT32=>UInt32,
+                      TF_INT64=>Int64, TF_UINT64=>UInt64,
+                      TF_FLOAT=>Float32, TF_DOUBLE=>Float64,
+                      TF_BOOL=>Bool, TF_STRING=>String,
+                      TF_COMPLEX64=>ComplexF32, TF_COMPLEX128=>ComplexF64)
 const inv_type_map = Dict(v=>k for (k, v) in type_map)
 
 function tf_to_jl_type(dt::TF_DataType)