Skip to content

Commit c4ae5b8

Browse files
Add missing type cases to methods as_numpy_name and get_datatype_size.
1 parent 4cab71f commit c4ae5b8

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/TensorFlowNET.Core/Tensors/dtypes.cs

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,10 @@ public static string as_numpy_name(this TF_DataType type)
188188
{
189189
TF_DataType.TF_STRING => "string",
190190
TF_DataType.TF_UINT8 => "uint8",
191+
TF_DataType.TF_INT8 => "int8",
192+
TF_DataType.TF_UINT32 => "uint32",
191193
TF_DataType.TF_INT32 => "int32",
194+
TF_DataType.TF_UINT64 => "uint64",
192195
TF_DataType.TF_INT64 => "int64",
193196
TF_DataType.TF_FLOAT => "float32",
194197
TF_DataType.TF_DOUBLE => "float64",
@@ -203,9 +206,12 @@ public static int get_datatype_size(this TF_DataType type)
203206
{
204207
TF_DataType.TF_BOOL => sizeof(bool),
205208
TF_DataType.TF_UINT8 => sizeof(byte),
206-
TF_DataType.TF_INT8 => sizeof(byte),
209+
TF_DataType.TF_INT8 => sizeof(sbyte),
210+
TF_DataType.TF_UINT16 => sizeof(ushort),
207211
TF_DataType.TF_INT16 => sizeof(short),
212+
TF_DataType.TF_UINT32 => sizeof(uint),
208213
TF_DataType.TF_INT32 => sizeof(int),
214+
TF_DataType.TF_UINT64 => sizeof(ulong),
209215
TF_DataType.TF_INT64 => sizeof(long),
210216
TF_DataType.TF_FLOAT => sizeof(float),
211217
TF_DataType.TF_DOUBLE => sizeof(double),

0 commit comments

Comments
 (0)