s="p">:</span> <span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd">    Function that implements :func:`~brevitas.function.ops.dpu_round` with a straight-through</span>
<span class="sd">    gradient estimator.</span>

<span class="sd">    Notes:</span>
<span class="sd">        Wrapper for either :func:`~brevitas.ops.autograd_ste_ops.dpu_ro