code: def _softmax_flops_compute(
      input, dim=None, _stacklevel=3, dtype=None
  ):
      return input.numel(), 0
  
imports: import time
  import torch
  import torch.nn
  import torch.nn.functional
  from functools import partial
  from typing import List, Optional
  from collections import OrderedDict
  import numpy
label_desc: Use image node_id 18 for calling a global function with example usage: _softmax_flops_compute(input, dim, _stacklevel, dtype) and returns: input,  int