Code

Implement cuDNN conv2D grad.#32

Code

Merged

feiwang3311 merged 5 commits intofeiwang3311:masterfeiwang3311/Lantern:masterfrom dan-zheng:cudnn-conv2ddan-zheng/Lantern:cudnn-conv2dCopy head branch name to clipboardOct 19, 2018

+397-127

Commits

Commits on Oct 18, 2018

• Implement conv2D grad on cudNN.

  Show description for ca56e6f
  dan-zheng

  committed
  ca56e6f

• Enable gradR_loss on GPU.

  Show description for 4d4a337
  dan-zheng

  committed
  4d4a337

• Implement TensorR transfer ops.

  Show description for dcb90ff
  dan-zheng

  committed
  dcb90ff

• Fix cuDNN conv2D_batch_grad, add test.

  Show description for 5ee53d4
  dan-zheng

  committed
  5ee53d4

Commits on Oct 19, 2018

• Merge branch 'master' into cudnn-conv2d

  dan-zheng

  authored
  535c1bb