ass="si">{</span><span class="n">direction</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">,</span>
                            <span class="n">size</span><span class="p">[</span><span class="n">i</span><span class="p">],</span>
                        <span class="p">)</span></div>

<div class="viewcode-block" id="TensorReplacer.make_megatron_layer"><a class="viewcode-back" hre