• X
    [cherry-pick] add abs,mean,sum,ge,gt,pow,etc higher-order differentiation operators (#46184) · ad8beaaf
    Xiaoxu Chen 提交于
    * [cherry-pick] extend reduce_sum,reduce_sum,eq,ne,ge,abs,pow,etc higher order operators
    
    * add reduce_mean,reduce_sum primitive ops
    * add ne_p gt_p primitive operators
    * add ge_p abs_p primitive oparators
    * add cast primitive operators
    * add pow,square prim2oirg rules
    * add elementwise_div orig2prim rule
    
    * [cherry-pick] add mean,sum,ge,gt,ne,abs,etc higher-order differentiation operators(#45888)
    
    * add reduce_mean,reduce_sum primitive ops
    
    * add ne_p gt_p primitive operators
    
    * add ge_p abs_p primitive oparators
    ad8beaaf
test_primops.py 6.1 KB