• X
    【prim】Layer norm (#50422) · 2f4763ee
    xiaoguoguo626807 提交于
    * fix composite mean op map
    
    * fix composite check output
    
    * init layer_norm
    
    * init layer_norm
    
    * map output from composite rule to origin op
    
    * add dropout op map
    
    * add input map check
    
    * polish log
    
    * modify rules
    
    * success test_forward
    
    * modify test without cinn
    
    * modify cinn test
    
    * modify cinn test
    
    * except fp64
    
    * except fp64
    
    * delete flatten
    
    * delete unused change
    
    * review
    
    * pass cpu test
    
    * code style
    
    * delete flatten fp16 error
    
    * modify flatten test
    
    ---------
    Co-authored-by: Ncyber-pioneer <chenzhuo@tju.edu.cn>
    2f4763ee
utils.py 7.8 KB