Allow specifying the default variable store to use instead of always creating a new dict.
Allow specifying the default variable store to use instead of always creating a new dict.

PiperOrigin-RevId: 421424750

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo