“8c7c53b3d5237bcdbcb42e492ec51bc581223549”上不存在“paddle/phi/kernels/gpu/reduce_sum_grad_kernel.cu”
  • 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
gather_op.h 2.2 KB