lass="o">.</span><span class="n">nb_iter</span><span class="p">)</span>

    <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">sanity_checks</span><span class="p">:</span>
      <span class="k">with</span> <span class="n">tf</span><span class="o">.</span><span class="n">control_dependencies</span><span class="p">(</span><span class="n">asserts</span><span class="p">):</span>
        <span class="n">adv_x</span> <span class="o">=</span> <span class="n">tf</span><sp