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