Tensorflow Dataset Too Large For Memory