om Policy object.&quot;</span>
        <span class="p">)</span>

        <span class="k">return</span> <span class="n">policy_cls</span></div>

<div class="viewcode-block" id="TensorReplacer.replace_modules"><a class="viewcode-back" href="../../../parallel/Replacing.html#parallelformers.parallel.replacing.TensorReplacer.replace_modules">[docs]</a>    <span class="k">def</span> <span class="nf">replace_modules</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
        <span c