Tensor: added missing data types

henon · henon · commit 5d16dc67cbd5 · 2019-05-10T22:34:35.000+02:00
diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.cs b/src/TensorFlowNET.Core/Tensors/Tensor.cs
@@ -191,6 +191,8 @@ public TF_DataType ToTFDataType(Type type)
                     return TF_DataType.TF_INT16;
                 case "Int32":
                     return TF_DataType.TF_INT32;
+                case "Int64":
+                    return TF_DataType.TF_INT64;
                 case "Single":
                     return TF_DataType.TF_FLOAT;
                 case "Double":
@@ -199,6 +201,8 @@ public TF_DataType ToTFDataType(Type type)
                     return TF_DataType.TF_UINT8;
                 case "String":
                     return TF_DataType.TF_STRING;
+                case "Boolean":
+                    return TF_DataType.TF_BOOL;
                 default:
                     throw new NotImplementedException("ToTFDataType error");
             }
