Preserve control outputs when saving and loading functions

I don't think this means anything is broken yet, but presumably would be a problem for Grappler turning on pruning by default. I ran into it when doing some tangential debugging.

PiperOrigin-RevId: 238528823