• Y
    Fix gather & stack op (#14355) · bd294378
    Yibing Liu 提交于
    * Add int type support for stack_op
    
    * Improve gather op to support index with shape N x 1
    
    test=develop
    
    * Fix stack_op kernel's registry
    
    test=develop
    bd294378
gather.cu.h 2.6 KB