Browse Source

Add missing type cases to methods as_numpy_name and get_datatype_size.

pull/866/head
darth-vader-lg 4 years ago
parent
commit
372b958dd7
No known key found for this signature in database GPG Key ID: 7034ED27746C0663
1 changed files with 7 additions and 1 deletions
  1. +7
    -1
      src/TensorFlowNET.Core/Tensors/dtypes.cs

+ 7
- 1
src/TensorFlowNET.Core/Tensors/dtypes.cs View File

@@ -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),


Loading…
Cancel
Save