code: def vocab_parallel_cross_entropy(
      vocab_parallel_logits, target, label_smoothing
  ):
      """Helper function for the cross entropy."""
      return _VocabParallelCrossEntropy.apply(
          vocab_parallel_logits, target, label_smoothing
      )
  
imports: import torch
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