[tf.data] Whitelist the "Assert", "Print", and "Timestamp" logging ops for iterator checkpointing.

These ops are marked as "stateful" because they are impure or have a side-effect, but they do not contain internal state. As such, functions using them are safe to serialize as part of an iterator checkpoint.

PiperOrigin-RevId: 210762896