“2aefb6ebcbc0bf7c1347a03cb02bb268a30272d9”上不存在“PaddleNLP/machine_translation/transformer/__init__.py”
  • Add topk op (#3760) · 3fbb692d
    武毅 提交于
    * init add
    
    * add topk op
    
    * someupdate
    
    * fix style check
    
    * add test py file
    
    * update top k cuda kernel
    
    * follow comments
    
    * remove debug print
    
    * fix casting error
    
    * fix casting error
    
    * fix casting error
    
    * fix rename bug...
    
    * fix travis
    3fbb692d
top_k_op.cc 2.6 KB