transformers

96d833b2 - Return scalar losses instead of per-sample means (#18013)

Return scalar losses instead of per-sample means (#18013) * Return scalar losses instead of per-sample means * Make loss shape (1,) instead of scalar * Allow scalar losses in test_loss_computation * Allow scalar losses in test_loss_computation * Allow scalar losses in test_loss_computation * Remove XLA loss function for RAG