code: def test_batch_mm_prohibited_mutation_if_node(self):
      @torch.jit.script
      def test_batch_mm(n: int, use_t1: bool):
          T1 = torch.zeros((n, n))
          T2 = torch.zeros((n, n))
          T3 = torch.zeros((n, n))
          T4 = torch.zeros((n, n))
          T5 = torch.zeros((n, n))
          T6 = torch.zeros((n, n))
          T7 = torch.zeros((n, n))
          T8 = torch.zeros((n, n))
          T9 = torch.zeros((n, n))
          T10 = torch.zeros((n, n))
          if use_t1:
              torch.relu_(T1)
              return (
                  torch.mm(T1, T2)
                  + torch.mm(T3, T4)
                  + torch.mm(T5, T6)
                  + torch.mm(T7, T8)
                  + torch.mm(T9, T10)
              )
          else:
              return (
                  torch.mm(T2, T3)
                  + torch.mm(T4, T5)
                  + torch.mm(T6, T7)
                  + torch.mm(T8, T9)
              )
  
      self.run_pass("batch_mm", test_batch_mm.graph)
      FileCheck().check_count("aten::mm", 5, exactly=True).check_count(
          "prim::MMTreeReduce", 1, exactly=True
      ).run(test_batch_mm.graph)
  
file_path: pytorch/test/jit/test_batch_mm.py
imports: import torch
  from torch.testing import FileCheck
  from torch.testing._internal.jit_utils import JitTestCase
label_desc: Use image node_id 6 for calling the TestBatchMM obj's underlying member method code with example usage: obj.test_batch_mm_prohibited_mutation_if_node() and returns: unknown, unknown