code: def _format_flops(flops: float) -> str:
      """Returns a formatted flops string"""
      if flops > 1e12:
          return f"{flops / 1e12:.2f} TFLOPs"
      elif flops > 1e9:
          return f"{flops / 1e9:.2f} GFLOPs"
      elif flops > 1e6:
          return f"{flops / 1e6:.2f} MFLOPs"
      elif flops > 1e3:
          return f"{flops / 1e3:.2f} kFLOPs"
      return f"{flops} FLOPs"
  
imports: from typing import Any, Callable, Dict, List, Tuple, Optional, Union
  from dataclasses import dataclass, field
  import torch
  import torch.fx
  from torch.fx import Interpreter
  from torch.fx.node import Argument, Node
  from .flop_utils import flop_count
label_desc: Use image node_id 1 for calling a global function with example usage: _format_flops(flops) and returns: flops, str, unknown, str, unknown, str, unknown, str, unknown, str