Don't transfer softmax loss to host