Fix passing the results of .cpu() to argument inputs.
#4923
| Job | Run time |
|---|---|
| 2m 27s | |
| 2m 27s |
.cpu() to argument inputs.
#4923
| Job | Run time |
|---|---|
| 2m 27s | |
| 2m 27s |