Change _model_submodule to build a proper (a proper subset of weights) submodule.
Change _model_submodule to build a proper (a proper subset of weights) submodule.

PiperOrigin-RevId: 454141103

tensorflow/gnn,tensorflow/gnn