From 372b958dd7eb2bbdd257ab3771091089e08abd6b Mon Sep 17 00:00:00 2001 From: darth-vader-lg Date: Wed, 1 Sep 2021 00:33:06 +0200 Subject: [PATCH] Add missing type cases to methods as_numpy_name and get_datatype_size. --- src/TensorFlowNET.Core/Tensors/dtypes.cs | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Core/Tensors/dtypes.cs b/src/TensorFlowNET.Core/Tensors/dtypes.cs index 7c3a291c..372ac676 100644 --- a/src/TensorFlowNET.Core/Tensors/dtypes.cs +++ b/src/TensorFlowNET.Core/Tensors/dtypes.cs @@ -189,7 +189,10 @@ namespace Tensorflow { TF_DataType.TF_STRING => "string", TF_DataType.TF_UINT8 => "uint8", + TF_DataType.TF_INT8 => "int8", + TF_DataType.TF_UINT32 => "uint32", TF_DataType.TF_INT32 => "int32", + TF_DataType.TF_UINT64 => "uint64", TF_DataType.TF_INT64 => "int64", TF_DataType.TF_FLOAT => "float32", TF_DataType.TF_DOUBLE => "float64", @@ -204,9 +207,12 @@ namespace Tensorflow { TF_DataType.TF_BOOL => sizeof(bool), TF_DataType.TF_UINT8 => sizeof(byte), - TF_DataType.TF_INT8 => sizeof(byte), + TF_DataType.TF_INT8 => sizeof(sbyte), + TF_DataType.TF_UINT16 => sizeof(ushort), TF_DataType.TF_INT16 => sizeof(short), + TF_DataType.TF_UINT32 => sizeof(uint), TF_DataType.TF_INT32 => sizeof(int), + TF_DataType.TF_UINT64 => sizeof(ulong), TF_DataType.TF_INT64 => sizeof(long), TF_DataType.TF_FLOAT => sizeof(float), TF_DataType.TF_DOUBLE => sizeof(double),