1. 02 11月, 2018 1 次提交
    • W
      Add affine grid generator op (#12238) · 0c319e0b
      whs 提交于
      * Add affine grid generator.
      
      * fix ffine grid.
      
      * Add unitest.
      
      * Add CPU kernel and fix unitest.
      
      * Fix CPU kernel.
      
      * Refine code.
      test=develop
      
      * Fix python api.
      test=develop
      
      * Update python api.
      test=develop
      
      * Fix comment.
      test=develop
      
      * Rename affine_grid_generator to affine_grid and enhence unitest.
      test=develop
      
      * Fix unitest.
      test=develop
      0c319e0b
  2. 29 10月, 2018 3 次提交
  3. 26 10月, 2018 1 次提交
  4. 16 10月, 2018 1 次提交
  5. 26 7月, 2018 1 次提交
  6. 06 6月, 2018 4 次提交
  7. 04 4月, 2018 1 次提交
  8. 19 3月, 2018 3 次提交
  9. 16 3月, 2018 2 次提交
  10. 15 3月, 2018 1 次提交
    • D
      [Speed]implement cudnn sequence softmax cudnn (#8978) · 128adf53
      dzhwinter 提交于
      * "add softmax cudnn functor support"
      
      * "add testing"
      
      * "refine cmakelist"
      
      * "sequence softmax forward speed up"
      
      * "add softmax grad"
      
      * "fix sequence softmax test"
      
      * "add double precision'
      
      * "fix softmax test"
      
      * "add softmax cudnn support"
      
      * "fix softmax cudnn test"
      
      * "add softmax to nn.py"
      
      * "fix compile bug"
      
      * "refine cmakelist"
      
      * "fix ci"
      
      * "fix based on comment"
      
      * "fix based on comments"
      
      * "fix ci"
      128adf53
  11. 07 3月, 2018 1 次提交
  12. 12 2月, 2018 1 次提交
  13. 10 2月, 2018 2 次提交
  14. 27 11月, 2017 1 次提交
    • Conv cudnn 3d (#5783) · a06bec12
      武毅 提交于
      * conv cudnn 3d
      
      * update test case
      
      * update
      
      * update
      
      * follow comments and remove groups from helper
      
      * update
      
      * refine
      
      * update
      
      * follow comments2
      
      * update
      
      * fix compile
      a06bec12
  15. 15 11月, 2017 1 次提交
  16. 13 11月, 2017 2 次提交
  17. 26 10月, 2017 1 次提交
    • Q
      Cudnn batch norm op (#5067) · 56b723c4
      Qiao Longfei 提交于
      * init cudnn batch norm op
      
      * rename batch_norm_cudnn_op.cc batch_norm_op.cu
      
      * correct name style
      
      * add ExtractNCWHD, simplify code
      
      * fix ExtractNCWHD
      
      * use CUDNN_ENFORCE instead of PADDLE_ENFORCE
      56b723c4
  18. 12 10月, 2017 1 次提交
    • Cudnn conv op (#4195) · a3ccbdb3
      武毅 提交于
      * add cudnn_conv_op
      
      * WIP
      
      * update
      
      * update
      
      * fix grad check
      
      * use platform::memory
      
      * add support group for cudnn
      
      * update
      
      * follow comments
      
      * fix onlycpu build
      
      * update cuda define
      
      * follow comments
      
      * follow comments
      
      * merge with updates
      
      * fix compile error
      
      * follow comments
      
      * follow comments
      a3ccbdb3
  19. 07 9月, 2017 1 次提交
  20. 04 9月, 2017 1 次提交
  21. 31 8月, 2017 2 次提交