pan class="n">calibrator</span><span class="p">;</span>
<span class="c1">/// Set a larger workspace (you may get better performace from doing so)</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">workspace_size</span> <span class="o">=</span> <span class="mi">1</span> <span class="o">&lt;&lt;</span> <span class="mi">28</span><span class="p">;</span>

<span class="k">auto</span> <span class="n">trt_mod</span> <span class="o">=</span> <span class="n">torch_tensorrt</span><span 