• Y
    Return parent_idx in beam_search op (#15520) · 16d54f7f
    Yiqun Liu 提交于
    * Refine beam_search_op to output an extra parent_idx tensor.
    test=develop
    
    * Fix the unittest test_beam_search_op.
    test=develop
    
    * Fix the merging mistake.
    test=develop
    16d54f7f
beam_search_op.h 2.2 KB