 <div class="highlight">
          <pre><span></span>[18]:
</pre>
         </div>
        </div>
        <div class="input_area highlight-ipython3 notranslate">
         <div class="highlight">
          <pre>
<span></span><span class="c1"># Obtain the average time taken by a batch of input</span>
<span class="n">benchmark</span><span class="p">(</span><span class="n">trt_model_fp16</span><span class="p">,</span> <span class="n">input_shape</span><span class="o">=</span><span class="p">(</span><span class="