• Fenglei's avatar
    gpu convolution support nd(n<4) (#824) · b9b7845c
    Fenglei authored
    * add convolution in progress
    
    * enable 1 test
    
    * convolution in progress
    
    * use filter descripter
    
    * filter discreptor bug fix
    
    * tensor format
    
    * add missed dimension calculator
    
    * forward convolution 4d without dilation and padding working
    
    * data dilation(deconvolution) and enable some test
    
    * add backprop convolution data and filter
    
    * backprop can compile
    
    * pass unit test, but still have problem on padding
    
    * 2d, symmtric padding, no data dilation works now
    
    * clean up code
    
    * extend gpu convolution to nd
    
    * fix some bugs
    
    * working version for upto 3d convolution, code format.
    
    * remove nunecessary changes
    
    * add restriction for data dilation and asymmetric padding
    
    * clang format
    
    * support upto 3D convolution for now
    
    * change comments to not implemented
    
    * change comments to not implemented
    
    * add quary for additional GPU workspace for convolution
    
    * clang format
    
    * code format
    
    * using row_major_strides
    
    * using join
    
    * fix bug for join
    
    * refactor dimension calculation
    b9b7845c
Name
Last commit
Last update
.ci/travis/ubuntu Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
third-party Loading commit data...
.clang-format Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.yml Loading commit data...
CMakeLists.txt Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
changes.md Loading commit data...