tf.saved_model: Allow saving/loading concrete functions directly

We'll want this for representing signatures, and I don't see any reason to limit it to that.

Some fiddling so that the function gets registered with the eager context. Apparently that was only working when graph building.

PiperOrigin-RevId: 229661863