ass="p">({</span><span class="n">input_shape</span><span class="p">});</span>

<span class="cp"># Set a precision. DLA supports fp16 or int8 only</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">enabled_precisions</span> <span class="o">=</span> <span class="p">{</span><span class="n">torch</span><span class="o">::</span><span class="n">kF16</span><span class="p">};</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</span><span class="p"