code: def py():
      for i in range(TIMES):
          flat = _flatten_dense_tensors(t)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/benchmarks/flatten_bench.py
imports: import argparse
  import gc
  import torch
  from torch._utils import _flatten_dense_tensors
  from deepspeed.ops.op_builder import UtilsBuilder
  from apex_C import flatten
  import cProfile
  import timeit
label_desc: Use image node_id 1 for calling a global function with example usage: py() without return types