code: def vocab_parallel_cross_entropy(
      vocab_parallel_logits, target, label_smoothing=0.0
  ):
      """Helper function for the cross entropy."""
      return _VocabParallelCrossEntropy.apply(
          vocab_parallel_logits, target, label_smoothing
      )
  
file_path: apex/apex/transformer/tensor_parallel/cross_entropy.py
imports: import torch
  from apex.transformer.parallel_state import get_tensor_model_parallel_group
  from apex.transformer.parallel_state import get_tensor_model_parallel_rank
  from apex.transformer.parallel_state import get_tensor_model_parallel_world_size
  from apex.transformer.tensor_parallel.utils import VocabUtility
label_desc: Use image node_id 1 for calling a global function with example usage: vocab_parallel_cross_entropy(vocab_parallel_logits, target, label_smoothing) and returns: _VocabParallelCrossEntropy