Handle nested tuples in GpuTransferManager.
This became necessary when the TOKEN primitive type was added. In some models, an existing tuple T is extended to (T, token[]). Also add the TOKEN case to a switch statement where it was missing. PiperOrigin-RevId: 202643759
Loading
Please sign in to comment