| @@ -191,6 +191,8 @@ namespace Tensorflow | |||||
| return TF_DataType.TF_INT16; | return TF_DataType.TF_INT16; | ||||
| case "Int32": | case "Int32": | ||||
| return TF_DataType.TF_INT32; | return TF_DataType.TF_INT32; | ||||
| case "Int64": | |||||
| return TF_DataType.TF_INT64; | |||||
| case "Single": | case "Single": | ||||
| return TF_DataType.TF_FLOAT; | return TF_DataType.TF_FLOAT; | ||||
| case "Double": | case "Double": | ||||
| @@ -199,6 +201,8 @@ namespace Tensorflow | |||||
| return TF_DataType.TF_UINT8; | return TF_DataType.TF_UINT8; | ||||
| case "String": | case "String": | ||||
| return TF_DataType.TF_STRING; | return TF_DataType.TF_STRING; | ||||
| case "Boolean": | |||||
| return TF_DataType.TF_BOOL; | |||||
| default: | default: | ||||
| throw new NotImplementedException("ToTFDataType error"); | throw new NotImplementedException("ToTFDataType error"); | ||||
| } | } | ||||