Add support to call `tf.saved_model.load` while in graph mode.

This works by wiring the restore ops of variables as their initializers.

For now there is only a small test added. Will follow up with changes in
load_test so most of the tests also exercise that code path.

PiperOrigin-RevId: 231940750