Fix tf memory leak: https://github.com/tensorflow/tensorflow/issues/3…