SIGN IN SIGN UP
apache / mxnet UNCLAIMED

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

0 0 20 C++

Improve convolution (#5284)

* Add im2col for convolution

* Implemented Forward function for convolution op

* Added req type to col2im functions

* Added backward for computing gradients wrt image data

* Added backward for weights and bias

* Fixed bugs and added one test case

* Added one test case

* Added infer shape for 1d conv

* Fixed a bug for computing gradient of bias

* Removed Convolution2

* Renamed old and new convolution implementations

* Fixed lint errors

* Fixed a lint error

* Pass stream to im2col gpu functions

* Fixed convolution n-D gpu bugs

* Added 1/2/3-D cpu/gpu convolution test cases

* Move im2col.cc and im2col.cu to header files

* Consolidated interfaces of im2col and col2im for cpu and gpu

* Clean up and add test cases

* Activated test case

* Fixed a typo

* Enable openMP for 2d image cpu convolution

* Reverted openmp

* Added mshadow change

* Fixed warnings of ConvertLayout in mshadow

* Fixed unsigned/signed integer comparison

* Fixed mkl build error on jenkins
R
reminisce committed
6b74dcd62a7c7608afd7a6499e38763e6ae8cbb9
Parent: 9203e38
Committed by Eric Junyuan Xie <piiswrong@users.noreply.github.com> on 3/17/2017, 7:42:11 PM