| 254 | } |
| 255 | |
| 256 | BiasedDecoder::BiasedDecoder(const float prefix_bias_beta, |
| 257 | const std::vector<std::vector<size_t>>& prefix_ids) |
| 258 | : _prefix_bias_beta(prefix_bias_beta) |
| 259 | , _prefix_ids(prefix_ids) |
| 260 | { |
| 261 | } |
| 262 | |
| 263 | void BiasedDecoder::decode(const dim_t cur_batch_size, |
| 264 | const size_t step, |
nothing calls this directly
no outgoing calls
no test coverage detected