Adding a whitelited_stateful_ops argument to the old Defun in which stateful...

Adding a whitelited_stateful_ops argument to the old Defun in which stateful ops belonging to this whitelist are recreated instead of being re-used when captured by value. This is useful for dataset make_one_shot_iterator which captures all inputs by value.

PiperOrigin-RevId: 224436333