code: def test_module_partitioner_functional_conv_relu_conv(self):
      class FunctionalConv2d(torch.nn.Module):
          def __init__(self):
              super().__init__()
              self.stride = (1, 1)
              self.padding = (0, 0)
              self.dilation = (1, 1)
              self.groups = 1
  
          def forward(self, x, weight, bias):
              return torch.nn.functional.conv2d(
                  x,
                  weight,
                  bias,
                  self.stride,
                  self.padding,
                  self.dilation,
                  self.groups,
              )
  
      class M(torch.nn.Module):
          def __init__(self):
              super().__init__()
              self.conv1 = FunctionalConv2d()
              self.conv2 = FunctionalConv2d()
  
          def forward(self, x, weight, bias):
              x = self.conv1(x, weight, bias)
              x = torch.nn.functional.relu(x)
              x = self.conv2(x, weight, bias)
              return x
  
      inputs = (
          torch.randn(1, 3, 5, 5),
          torch.rand(3, 3, 3, 3),
          torch.rand(3),
      )
      gm, _ = torch._dynamo.export(M(), aten_graph=True)(*inputs)
      gm.graph.eliminate_dead_code()
  
      module_partitions = get_source_partitions(
          gm.graph, [torch.nn.functional.conv2d]
      )
  
      self.assertEqual(len(module_partitions), 1)
      self.assertEqual(
          len(module_partitions[torch.nn.functional.conv2d]), 2
      )
  
file_path: pytorch/test/fx/test_source_matcher_utils.py
imports: import os
  import sys
  import unittest
  import torch
  from torch._dynamo.eval_frame import is_dynamo_supported
  from torch.fx.passes.utils.source_matcher_utils import get_source_partitions, check_subgraphs_connected
  from torch.testing._internal.jit_utils import JitTestCase
label_desc: Use image node_id 3 for calling the TestSourceMatcher obj's underlying member method code with example usage: obj.test_module_partitioner_functional_conv_relu_conv() and returns: torch, x