Browse Source
Add missing type cases to methods as_numpy_name and get_datatype_size.
tags/TensorFlowOpLayer
darth-vader-lg
Esther Hu
4 years ago
1 changed files with 7 additions and 1 deletions
Unified View
Diff Options
+ 7
- 1
src/TensorFlowNET.Core/Tensors/dtypes.cs
View File
| @@ -189,7 +189,10 @@ namespace Tensorflow | |||||
| { | { | ||||
| TF_DataType.TF_STRING => "string", | TF_DataType.TF_STRING => "string", | ||||
| TF_DataType.TF_UINT8 => "uint8", | TF_DataType.TF_UINT8 => "uint8", | ||||
| TF_DataType.TF_INT8 => "int8", | |||||
| TF_DataType.TF_UINT32 => "uint32", | |||||
| TF_DataType.TF_INT32 => "int32", | TF_DataType.TF_INT32 => "int32", | ||||
| TF_DataType.TF_UINT64 => "uint64", | |||||
| TF_DataType.TF_INT64 => "int64", | TF_DataType.TF_INT64 => "int64", | ||||
| TF_DataType.TF_FLOAT => "float32", | TF_DataType.TF_FLOAT => "float32", | ||||
| TF_DataType.TF_DOUBLE => "float64", | TF_DataType.TF_DOUBLE => "float64", | ||||
| @@ -204,9 +207,12 @@ namespace Tensorflow | |||||
| { | { | ||||
| TF_DataType.TF_BOOL => sizeof(bool), | TF_DataType.TF_BOOL => sizeof(bool), | ||||
| TF_DataType.TF_UINT8 => sizeof(byte), | 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_INT16 => sizeof(short), | ||||
| TF_DataType.TF_UINT32 => sizeof(uint), | |||||
| TF_DataType.TF_INT32 => sizeof(int), | TF_DataType.TF_INT32 => sizeof(int), | ||||
| TF_DataType.TF_UINT64 => sizeof(ulong), | |||||
| TF_DataType.TF_INT64 => sizeof(long), | TF_DataType.TF_INT64 => sizeof(long), | ||||
| TF_DataType.TF_FLOAT => sizeof(float), | TF_DataType.TF_FLOAT => sizeof(float), | ||||
| TF_DataType.TF_DOUBLE => sizeof(double), | TF_DataType.TF_DOUBLE => sizeof(double), | ||||