pan>
                    <span class="n">trace_func</span><span class="o">=</span><span class="k">lambda</span> <span class="n">module</span><span class="p">,</span> <span class="n">inputs</span><span class="p">:</span> <span class="n">acc_tracer</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span>
                        <span class="n">module</span><span class="p">,</span>
                        <span class="n">inputs</span><span class="p">,</span>  <span class="c1"># type: 