Fix rewrap to always return the updated entity. Use it as such at the autograph integration point. In turn, temporarily disable autograph for dataset functions until we reduce the conversion overhead.
Fixes #25159.
With thanks to @ktaebum for identifying this fix.

PiperOrigin-RevId: 235255035