| 101 | } |
| 102 | |
| 103 | int paddle_beam_decode(at::Tensor th_probs, |
| 104 | at::Tensor th_seq_lens, |
| 105 | std::vector<std::string> labels, |
| 106 | int vocab_size, |
| 107 | size_t beam_size, |
| 108 | size_t num_processes, |
| 109 | double cutoff_prob, |
| 110 | size_t cutoff_top_n, |
| 111 | size_t blank_id, |
| 112 | int log_input, |
| 113 | at::Tensor th_output, |
| 114 | at::Tensor th_timesteps, |
| 115 | at::Tensor th_scores, |
| 116 | at::Tensor th_out_length){ |
| 117 | |
| 118 | return beam_decode(th_probs, th_seq_lens, labels, vocab_size, beam_size, num_processes, |
| 119 | cutoff_prob, cutoff_top_n, blank_id, log_input, NULL, th_output, th_timesteps, th_scores, th_out_length); |
| 120 | } |
| 121 | |
| 122 | int paddle_beam_decode_lm(at::Tensor th_probs, |
| 123 | at::Tensor th_seq_lens, |
nothing calls this directly
no test coverage detected