somaticio / tensorflow.rb

tensorflow for ruby
BSD 3-Clause "New" or "Revised" License
829 stars 63 forks source link

Fix the memory leak in Graph Importing #96

Closed AshtonO closed 7 years ago

AshtonO commented 7 years ago

The C++ bridge code for graph importing would leak the data buffer that's used to import the graph. This fix implements the proper buffer deallocator function and calls delete buffer once a graph is loaded.

arafatkatze commented 7 years ago

@AshtonO Thanks a lot.