Browse Source

output_names exeption if length is zero.

tags/v0.100.5-BERT-load
Haiping Chen 2 years ago
parent
commit
f4450e9e53
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      src/TensorFlowNET.Core/Graphs/FuncGraph.cs

+ 5
- 0
src/TensorFlowNET.Core/Graphs/FuncGraph.cs View File

@@ -61,6 +61,11 @@ public class FuncGraph : Graph, IDisposable
string[] output_names)
{
var status = new Status();
if (output_names != null && output_names.Length == 0)
{
output_names = null;
};

_func_graph_handle = c_api.TF_GraphToFunction(_handle,
_graph_key,
false,


Loading…
Cancel
Save