class_docstr: Divide tokens equally among the tensor parallel ranks
class_docstr_tok: ["Divide","tokens","equally","among","the","tensor","parallel","ranks"]
code: def symbolic(graph, input_, dim):
      return _drop_tokens(input_, dim)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/moe/mappings.py
imports: import torch
  import deepspeed
label_desc: Use image node_id 1 for calling the _DropTokens obj's underlying member method code with example usage: obj.symbolic(graph, input_, dim) and returns: _drop_tokens