span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>

<span class="cp"># Set the DLA core id</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</span><span class="p">.</span><span class="n">dla_core</span> <span class="o">=</span> <span class="mi">1</span><span class="p">;</span>

<span class="cp"># If a layer fails to run on DLA it will fallback to GPU</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</