tf.data.experimental.assert_cardinality