transformers

Create the return value on device to avoid unnecessary copying from CPU

#26151

Merged