Browse Source

CNN Text Classification passed

tags/v0.12
Oceania2018 6 years ago
parent
commit
f740b1a239
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      src/TensorFlowNET.Core/Operations/OpDefLibrary.cs

+ 2
- 0
src/TensorFlowNET.Core/Operations/OpDefLibrary.cs View File

@@ -141,6 +141,8 @@ namespace Tensorflow
dtype = input_arg.Type; dtype = input_arg.Type;
else if (attrs.ContainsKey(input_arg.TypeAttr)) else if (attrs.ContainsKey(input_arg.TypeAttr))
dtype = (DataType)attrs[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)) else if (default_type_attr_map.ContainsKey(input_arg.TypeAttr))
default_dtype = (DataType)default_type_attr_map[input_arg.TypeAttr]; default_dtype = (DataType)default_type_attr_map[input_arg.TypeAttr];




Loading…
Cancel
Save