Fix passing the results of .cpu() to argument inputs.
#1911
| Job | Run time |
|---|---|
| 48s | |
| 52s | |
| 1m 40s |
.cpu() to argument inputs.
#1911
| Job | Run time |
|---|---|
| 48s | |
| 52s | |
| 1m 40s |