Implement cuDNN conv2D grad. - Githubissues

feiwang3311 / Lantern

BSD 3-Clause "New" or "Revised" License

167 stars 15 forks source link

Implement cuDNN conv2D grad. #32

Closed dan-zheng closed 5 years ago

dan-zheng commented 5 years ago

Make conv2D_batch_grad backend-defined.
Implement cuDNN conv2D_batch_grad.
Enable gradR_loss on GPU.
(Not related) Implement TensorR transfer ops.

Conv2D is fully implemented for cuDNN.