“158bf13f1c133c6af77674560e33413be552d51f”上不存在“paddle/phi/kernels/gpu/eigh_grad_kernel.cu”
  • 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.cu 9.6 KB