1. 23 7月, 2019 1 次提交
    • W
      Cudnn convolution reconstruction (#18284) · 6b78e00d
      wangchaochaohu 提交于
      * rewrite the conv_op using cudnn_conv_helper
      
      * add workspace limit for v7 test=develop
      
      * fix test=develop
      
      * add half float test=develop
      
      * fix test=develop
      
      * fix test=develop
      
      * revise code style test=develop
      
      * fix test=develop
      6b78e00d
  2. 10 5月, 2019 1 次提交
    • Q
      Double backward of conv2d. (#17211) · e32c9888
      qingqing01 提交于
      * Add conv2d_grad_grad_op
      * Extracte the cuDNN conv algo searching code in conv_cudnn_helper.h.
          - Now use it in conv2d_grad_grad.
          - Will simply the searching code in conv2d and conv2d_grad in next PR.
      * Enhance and fix bug in unit testing of gradient_checker.
      * Support to fetch empty variables,return None in Python.
      e32c9888