span> <span class="o">==</span> <span class="n">torch</span><span class="o">.</span><span class="n">float32</span>
                        <span class="k">else</span> <span class="n">x</span>
                    <span class="p">)</span>
                <span class="k">else</span><span class="p">:</span>
                    <span class="n">conversion_fn</span> <span class="o">=</span> <span class="n">fp16_conversion_fn</span>

                <span class="n">inputs</span> <span class="o">=</span> <span class