Handling large data in TensorFlow