[pytorch] Unify device placement and no_grad

for pytorch models
6 jobs for fix-pytorch-models in 59 minutes (queued for 6 seconds)