(x)
| 194 | input_flatten = lambda x: nest.flatten(x) if input_is_sequence else [x] |
| 195 | |
| 196 | def input_pack(x): |
| 197 | return nest.pack_sequence_as(elems, x) if input_is_sequence else x[0] |
| 198 | |
| 199 | elems_flat = input_flatten(elems) |
| 200 | elems_flat = ragged_tensor.match_row_splits_dtypes(*elems_flat) |