• Y
    Add group_norm composite rule (#51874) · cabf3921
    Yichen Zhang 提交于
    * add group_norm composite rule
    
    * add test for scale_grad and bias_grad
    
    * resolve conflicts
    
    * remove amp in composite_rule.py
    
    * add float16 test
    
    * deal with NHWC format
    
    * keep the composite rule in float16 identical as original kernel
    
    * resolve conflicts
    cabf3921
composite_rules.py 18.3 KB