• Z
    Add mean composite rule (#50298) · f7f67b72
    zqw_1997 提交于
    * beta
    
    * small commit
    
    * add batch_norm composite rule
    
    move composite test case
    
    remove unuseful var
    
    add composite op blacklist
    
    * small change v2
    
    * finish the test_composite_mean and test_composite_mean_grad
    
    * add ops assertion to the tests
    
    * add cinn test
    
    * fix the error and inappropriate usage in func: mean_composite
    
    * remove the ref of outer lib in primtives.py
    
    * modify sample code of reduce_sum
    
    * fix composite mean op map
    
    * modify testcases to test more float type
    
    * remove cpu float16 test
    
    * cinn test fix
    
    * remove reduce_max
    
    * change the name sum to sum_x
    
    * change the use of reduce_sum to sum
    
    ---------
    Co-authored-by: Ncyber-pioneer <chenzhuo@tju.edu.cn>
    f7f67b72
nn.py 28.1 KB