Correctly extract dtype for tuples in to_streaming_enference
Correctly extract dtype for tuples in to_streaming_enference

PiperOrigin-RevId: 463388744

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