• Z
    Add dscending for argsort (#21400) · b1627455
    zhaoyuchen2018 提交于
    * Add ascending for argsort
    
    * Refine api doc description.
    
    * Refine descending description
    
    * Add int32 logic to speedup when data is small size.
    
    * Remove int32 opt as not support in python
    b1627455
multihead_matmul_op.cu 14.2 KB