Tensorflow Data Pipeline Management