e><span></span>%ret : Tensor = prim::If(%622)
block0():
  %ret.1 : Tensor = aten::addmm(%self.fc.bias, %x9.1, %3677, %3, %3)
  -&gt; (%ret.1)
block1():
  %output.1 : Tensor = aten::matmul(%x9.1, %3677)
  %output0.1 : Tensor = aten::add_(%output.1, %self.fc.bias, %3)
  -&gt; (%output0.1)
</pre>
        </div>
       </div>
       <p>
        We fuse this set of blocks into a graph like this:
       </p>
       <div class="highlight-none notranslate">
        <div class="highlight">
         <pre><span></span