Browse Source

disable TF_UpdateEdge.

tags/v0.30
Oceania2018 5 years ago
parent
commit
69967b4812
2 changed files with 4 additions and 1 deletions
  1. +3
    -1
      src/TensorFlowNET.Core/Operations/Operation.cs
  2. +1
    -0
      test/TensorFlowNET.UnitTest/Tensorflow.UnitTest.csproj

+ 3
- 1
src/TensorFlowNET.Core/Operations/Operation.cs View File

@@ -322,11 +322,13 @@ namespace Tensorflow

// Reset cached inputs.
_inputs_val = null;
_node_def = null;
// after the c_api call next time _inputs is accessed
// the updated inputs are reloaded from the c_api
lock (Locks.ProcessWide)
{
c_api.TF_UpdateEdge(_graph, output, input, tf.Status.Handle);
// disable
// c_api.TF_UpdateEdge(_graph, output, input, tf.Status.Handle);
//var updated_inputs = inputs;
tf.Status.Check();
}


+ 1
- 0
test/TensorFlowNET.UnitTest/Tensorflow.UnitTest.csproj View File

@@ -47,6 +47,7 @@

<ItemGroup>
<PackageReference Include="FluentAssertions" Version="5.10.3" />
<PackageReference Include="MethodBoundaryAspect.Fody" Version="2.0.138" />
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.7.1" />
<PackageReference Include="MSTest.TestAdapter" Version="2.1.2" />
<PackageReference Include="MSTest.TestFramework" Version="2.1.2" />


Loading…
Cancel
Save