From 5d16dc67cbd5c4182cb946b3c11394004cf1455a Mon Sep 17 00:00:00 2001 From: Meinrad Recheis Date: Fri, 10 May 2019 22:34:35 +0200 Subject: [PATCH] Tensor: added missing data types --- src/TensorFlowNET.Core/Tensors/Tensor.cs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/TensorFlowNET.Core/Tensors/Tensor.cs b/src/TensorFlowNET.Core/Tensors/Tensor.cs index a4cc1769..34b98b35 100644 --- a/src/TensorFlowNET.Core/Tensors/Tensor.cs +++ b/src/TensorFlowNET.Core/Tensors/Tensor.cs @@ -191,6 +191,8 @@ namespace Tensorflow return TF_DataType.TF_INT16; case "Int32": return TF_DataType.TF_INT32; + case "Int64": + return TF_DataType.TF_INT64; case "Single": return TF_DataType.TF_FLOAT; case "Double": @@ -199,6 +201,8 @@ namespace Tensorflow return TF_DataType.TF_UINT8; case "String": return TF_DataType.TF_STRING; + case "Boolean": + return TF_DataType.TF_BOOL; default: throw new NotImplementedException("ToTFDataType error"); }