- 15 9月, 2021 1 次提交
-
-
由 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
-
- 20 5月, 2020 1 次提交
-
-
由 jinyuKING 提交于
API/OP(sequence_first_step, sequence_last_step, sequence_mask, beam_search, beam_search_decode) error message enhancement (#24590)
-
- 08 3月, 2019 2 次提交
- 30 1月, 2019 1 次提交
-
-
由 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
-
- 24 1月, 2019 1 次提交
-
-
由 Yiqun Liu 提交于
* Refine the beam_search op and test. * A basic CUDA implementation of beam_search for small batch_size. * Implement CUDA kernel for beam_search_op. * Use multiple CUDA threads in the same block to select the top beam. * Update the python api of beam_search op. * Enable extend function in CPU kernel of beam_search op. * Unify the CUDA codes. test=develop * Unify the CPU kernel of beam_search op. * Ensure the seletced items of beam_search_op's CPU kernel sorted by scores. * Update the description of beam_search in API.spec. * Enable the use of CUDA kernel in beam_search op. * Exclude the beam_search's CUDA unittest when there is no CUDA gpu, and delete some debuging statements. test=develop * Follow comments. test=develop * Call the CPU kernel for beam_search op when batch_size > 4. test=develop * Remove the except of is_empty op in PrepareData. test=develop
-
- 07 6月, 2018 1 次提交
-
-
由 guosheng 提交于
-
- 06 6月, 2018 1 次提交
-
-
由 guosheng 提交于
-
- 21 5月, 2018 1 次提交
-
-
由 dzhwinter 提交于
-
- 28 4月, 2018 1 次提交
-
-
由 ktlichkid 提交于
-
- 27 4月, 2018 1 次提交
-
-
由 ktlichkid 提交于
-
- 23 4月, 2018 2 次提交
- 20 4月, 2018 1 次提交
-
-
由 ktlichkid 提交于
-
- 19 4月, 2018 3 次提交
- 18 4月, 2018 2 次提交
-
-
由 ktlichkid 提交于
-
由 Abhinav Arora 提交于
* Fix cpplint warnings in beam_search_decode_op * Fix cpplint warnings in beam_search_op * Fix test * fix
-
- 12 2月, 2018 1 次提交
-
-
由 qingqing01 提交于
-
- 10 2月, 2018 2 次提交
- 09 2月, 2018 1 次提交
-
-
由 Yang Yang 提交于
-
- 23 1月, 2018 1 次提交
-
-
由 Qiao Longfei 提交于
neural machine translation model support beam search with while op
-
- 18 1月, 2018 1 次提交
-
-
由 Yan Chunwei 提交于
-
- 24 12月, 2017 1 次提交
-
-
由 dzhwinter 提交于
* "change operator interface" * "move devicepool to device_context" * "fix operator test" * "fix op_registry Run interface" * "net op passed. Need to fix nccl multi-Context" * "add nccl group function" * "add nccl group function" * "fix gpu count exceed 32 error" * "fix recurrent op, nccl op" * "change the other operators interface with Place" * "fix typo" * "fix pybind" * "fix device in python side" * "fix pybind failed" * "add init for test" * "fix CI"
-
- 15 11月, 2017 1 次提交
-
-
由 Yan Chunwei 提交于
-