“e9e8b46c505c902bd837be6e01fc295b36dc199b”上不存在“projects/zadam/imports.yml”
  • 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