feat: improve how device switch is handled between the metric device and the input tensors device #4698
| Job | Run time |
|---|---|
| 1m 19s | |
| 1m 12s | |
| 1m 19s | |
| 1m 17s | |
| 1m 9s | |
| 1m 19s | |
| 1m 7s | |
| 1m 19s | |
| 1m 15s | |
| 1m 14s | |
| 12m 30s |
| Job | Run time |
|---|---|
| 1m 19s | |
| 1m 12s | |
| 1m 19s | |
| 1m 17s | |
| 1m 9s | |
| 1m 19s | |
| 1m 7s | |
| 1m 19s | |
| 1m 15s | |
| 1m 14s | |
| 12m 30s |