fairseq/criterions/cross_entropy.py
fairseq/model_parallel/criterions/vocab_parallel_cross_entropy.py

fairseq/criterions/hubert_criterion.py
fairseq/model_parallel/criterions/vocab_parallel_cross_entropy.py

fairseq/criterions/adaptive_loss.py
fairseq/model_parallel/criterions/vocab_parallel_cross_entropy.py

fairseq/criterions/sentence_ranking.py
fairseq/model_parallel/criterions/vocab_parallel_cross_entropy.py

fairseq/criterions/masked_lm.py
fairseq/model_parallel/criterions/vocab_parallel_cross_entropy.py