code: def test_trace_with_inductor_decomp(self):
      model = SimpleMLP()
      args = (torch.randn(8, 50),)
      kwargs = {}
      graphs = get_inductor_decomp_graphs(model, args, kwargs)
      assert len(graphs) == 2, "Expect fwd + bwd graphs"
      self.assertIsInstance(graphs[0], torch.fx.GraphModule)
      self.assertIsInstance(graphs[1], torch.fx.GraphModule)
  
file_path: pytorch/test/distributed/_tensor/debug/test_op_coverage.py
imports: import torch
  import torch.nn
  from torch.distributed._tensor.debug.op_coverage import get_inductor_decomp_graphs
  from torch.testing._internal.common_utils import run_tests, TestCase
label_desc: Use image node_id 1 for calling the TestOpCoverage obj's underlying member method code with example usage: obj.test_trace_with_inductor_decomp() without return types