PiperOrigin-RevId: 343746154 Change-Id: I241e0c951c46086e9f043588ddb3aa0eff23e593
tf.data.Dataset.window
drop_remainder=True
tf.data.experimental.AUTOTUNE