code: def conv_flop_jit(inputs: List[Any], outputs: List[Any]):
      """
      Count flops for convolution.
      """
      x, w = inputs[:2]
      x_shape, w_shape, out_shape = (x.shape, w.shape, outputs[0].shape)
      transposed = inputs[6]
  
      return conv_flop_count(
          x_shape, w_shape, out_shape, transposed=transposed
      )
  
file_path: colossalai/colossalai/fx/profiler/opcount.py
imports: import operator
  from functools import partial, reduce
  from numbers import Number
  from typing import Any, Callable, List
  import torch
  from packaging import version
label_desc: Use image node_id 7 for calling a global function with example usage: conv_flop_jit(inputs, outputs) and returns: conv_flop_count