class_docstr: All gather tokens among the tensor parallel ranks
code: def symbolic(graph, input_, dim):
      return _gather_tokens(input_, dim)
  
file_path: DeepSpeed/deepspeed/moe/mappings.py
imports: import torch
  import deepspeed
label_desc: Use image node_id 1 for calling the _GatherTokens obj's underlying member method code with example usage: obj.symbolic(graph, input_, dim) and returns: _gather_tokens