• 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
stack_op.cu 1.4 KB