[tf.data] More robust solution for input pipeline <--> performance model coordination.