From f740b1a2395c8a3c9ad36f6b313eed19e478f59d Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Fri, 16 Aug 2019 08:01:54 -0500 Subject: [PATCH] CNN Text Classification passed --- src/TensorFlowNET.Core/Operations/OpDefLibrary.cs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/TensorFlowNET.Core/Operations/OpDefLibrary.cs b/src/TensorFlowNET.Core/Operations/OpDefLibrary.cs index ab34a320..934ae7d2 100644 --- a/src/TensorFlowNET.Core/Operations/OpDefLibrary.cs +++ b/src/TensorFlowNET.Core/Operations/OpDefLibrary.cs @@ -141,6 +141,8 @@ namespace Tensorflow dtype = input_arg.Type; else if (attrs.ContainsKey(input_arg.TypeAttr)) dtype = (DataType)attrs[input_arg.TypeAttr]; + else if (values.GetType() == typeof(string) && dtype == DataType.DtInvalid) + dtype = DataType.DtString; else if (default_type_attr_map.ContainsKey(input_arg.TypeAttr)) default_dtype = (DataType)default_type_attr_map[input_arg.TypeAttr];