Tensorflow 2 reset graph