Which one is faster implementation ?
Having criterion = nn.CrossEntropyLoss()
but calling loss using predications (gpu tensor) and target (gpu tensor),

criterion(predications, target)
criterion(predications.cpu(), target.cpu())

Which of the above 2 is faster or both perform the same speed ?