1. 07 9月, 2020 6 次提交
  2. 06 9月, 2020 2 次提交
  3. 05 9月, 2020 1 次提交
  4. 04 9月, 2020 5 次提交
  5. 03 9月, 2020 5 次提交
  6. 02 9月, 2020 10 次提交
  7. 01 9月, 2020 4 次提交
    • Z
      [Paddle-TRT] Stack op plugin (#25605) · ad6e3dd6
      zlsh80826 提交于
      * add stack_op to CMakeLists
      
      * add dim=3 support for scale op
      
      * add trt stack op, test=develop
      
      * remove debug message
      
      * add stack plugin serialize
      
      * remove slice, scale op, will add later
      
      * enhence error message
      
      * revise trt ernie test to conver the stack op CI testi, test=develop
      
      * add stack op serialization
      
      * fix test shape after adding stack op
      
      * remove slice op, will add after implementing serialization
      
      * roll back to min_graph=5 to avoid using slice op
      
      * fix scale op output layer
      
      * implement stack op createPlugin
      
      * use workspace and move the defination to .cu
      
      * move stack plugin creator definition to .cu, test=develop
      ad6e3dd6
    • L
      Refine bernoulli and unsqueeze op (#26842) · 60ffc220
      Leo Chen 提交于
      * add check for bernoulli and register bool for unsqueeze
      
      * follow comments
      60ffc220
    • Revert "Add mkldnn bfloat16 option to C-API (#26676)" (#26854) · ced6e87e
      石晓伟 提交于
      This reverts commit 02083bda.
      ced6e87e
    • T
      add embedding 2.0 (#26649) · ebc5f997
      tangwei12 提交于
      * add embedding 2.0
      
      * add embedding support input int32
      ebc5f997
  8. 31 8月, 2020 7 次提交