code: def forward(self, x1, x2):
      x1 = x1 * 2
      x1 = self.linear1(x1)
      x1 = self.relu(x1)
      x1 = self.linear2(x1)
      out = x1 + x2
      return out
  
file_path: ColossalAI/tests/test_auto_parallel/test_tensor_shard/test_liveness_analysis.py
imports: import pytest
  import torch
  import torch.nn
  from colossalai._analyzer.fx.graph_module import ColoGraphModule
  from colossalai._analyzer.fx.passes import shape_prop_pass
  from colossalai._analyzer.fx.tracer.tracer import ColoTracer
  from colossalai.auto_parallel.tensor_shard.solver import GraphAnalyser
  from colossalai.fx import ColoGraphModule, ColoTracer
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 2 for calling the LinearModel obj's underlying member method code with example usage: obj.forward(x1, x2) and returns: out