• S
    add dynamic layer_norm plugin (#33293) · 45d1ae21
    Shang Zhizhou 提交于
    * add dynamic layer_norm plugin
    
    * fix bug
    
    * fix numpy.allclose
    
    * fix format
    
    * fix code style
    
    * remove shepe in dynamic shape
    
    * code format
    
    * remove layer norm fp16
    
    * fix format
    45d1ae21
layer_norm_op.cu 41.8 KB