Comment by tcdent

Where in the codebase is the logic specific to TPU vs. CUDA?