Browse Source

fix unit test.

tags/keras_v0.3.0
Oceania2018 4 years ago
parent
commit
78905e3864
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Functions/ConcreteFunction.cs
  2. +1
    -1
      src/TensorFlowNET.Core/Graphs/AutoGraphAttribute.cs

+ 1
- 1
src/TensorFlowNET.Core/Functions/ConcreteFunction.cs View File

@@ -40,7 +40,7 @@ namespace Tensorflow.Functions
{
func_graph = graph;

ToGraph(graph.Inputs, graph.Outputs);
ToGraph(graph.Inputs, graph.Outputs.Where(x => x != null).ToArray());
}

public ConcreteFunction(Func<Tensor, Tensor> func, TF_DataType dtype)


+ 1
- 1
src/TensorFlowNET.Core/Graphs/AutoGraphAttribute.cs View File

@@ -18,7 +18,7 @@ namespace Tensorflow.Graphs

public override void OnEntry(MethodExecutionArgs args)
{
func_name = $"autograph_{args.Instance.GetHashCode()}.{args.Method.Name}";
func_name = $"autograph_{args.Instance.GetType().FullName}.{args.Method.Name}";

if (functions.ContainsKey(func_name))
{


Loading…
Cancel
Save