Browse Source

add DisposableObject as base class.

tags/v0.12
Oceania2018 6 years ago
parent
commit
abdf4d815b
1 changed files with 18 additions and 0 deletions
  1. +18
    -0
      src/TensorFlowNET.Core/Sessions/BaseSession.cs

+ 18
- 0
src/TensorFlowNET.Core/Sessions/BaseSession.cs View File

@@ -48,6 +48,10 @@ namespace Tensorflow
_session = c_api.TF_NewSession(_graph, opts ?? newOpts, status);
// dispose newOpts
if (opts == null)
c_api.TF_DeleteSessionOptions(newOpts);
status.Check(true);
}
@@ -373,5 +377,19 @@ namespace Tensorflow
status.Check(true);
}
}
public void close()
{
Dispose();
}
protected override void DisposeUnManagedState()
{
using (var status = new Status())
{
c_api.TF_DeleteSession(_handle, status);
status.Check(true);
}
}
}
}

Loading…
Cancel
Save