Make `_linear` function public in `models.py`.
Make `_linear` function public in `models.py`.

PiperOrigin-RevId: 328667759

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research