• C
    [Pten] Add reduce mean kernel, replace with mean API (#37559) · f9e9fd19
    chentianyu03 提交于
    * add pten reduce kernel
    
    * add reduce_sum kernel
    
    * update attribute args and order
    
    * make out dtype undefined
    
    * fix empty input error
    
    * merge develop branch
    
    * rename sum as reduce function
    
    * rename sum as reduce function
    
    * fix reducekernelImpl args error
    
    * add reduce cuda kernel
    
    * modify dims type to const &
    
    * remove unsed log
    
    * fix reduce_all out eigen function error
    
    * remove unused codes
    
    * add the missing sum api define and testcase
    
    * merge develop branch
    
    * fix sum test axis value error
    
    * replace pten mean kernel with reduce_mean
    
    * revcover meam cuda to original implement
    f9e9fd19
test_mean_dev_api.cc 2.2 KB