Add multiple inputs support as list (in addition to tuple).
Add multiple inputs support as list (in addition to tuple).

PiperOrigin-RevId: 485622836

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