• P
    [NPU] add beam_search npu op (#34860) · 3760be06
    pangyoki 提交于
    * add beam_search npu op
    
    * fix CMakeList and add unittest
    
    * fix bug of beam search npu op
    
    * fix unittest
    
    * let input ids become int64
    
    * set output ids to int64_t
    
    * delete check_dygraph
    
    * fix beam_width=1
    3760be06
test_beam_search_op.py 14.1 KB