Cross entropy loss considers all your classes during training/evaluation.
Argmax is used only to get the class prediction (the class with the highest probability), this is used only during inference, not training/evaluation.