code: def count_call(
      module: torch.fx.GraphModule, op: str, target_op: Any
  ) -> int:
      return sum(
          [
              1 if (n.op == op and n.target == target_op) else 0
              for n in module.graph.nodes
          ]
      )
  
file_path: pytorch/test/inductor/test_fx_fusion.py
imports: from typing import Any, Callable
  import torch
  from torch._inductor.fx_passes.pre_grad import linear_permute_fusion, linear_transpose, permute_linear_fusion, permute_matmul_fusion, sink_cat_after_pointwise, transpose_linear, transpose_matmul
  from torch.fx.passes.shape_prop import ShapeProp
  from torch.testing._internal.common_utils import run_tests, TestCase
label_desc: Use image node_id 2 for calling a global function with example usage: count_call(module, op, target_op) and returns: sum