Commit ce979f79 authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

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
parent e0af631f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment