• 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
..
files Loading commit data...
models Loading commit data...
ref_generators Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
autodiff.in.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_performance.cpp Loading commit data...
backend_test.in.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
codegen.cpp Loading commit data...
convolution_test.in.cpp Loading commit data...
copy.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpio.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cudnn.cpp Loading commit data...
distributed.cpp Loading commit data...
eigen.cpp Loading commit data...
element_type.cpp Loading commit data...
file_util.cpp Loading commit data...
graph_partition.cpp Loading commit data...
includes.cpp Loading commit data...
inliner.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
mkldnn.cpp Loading commit data...
ngraph.cpp Loading commit data...
op.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pattern.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
runtime_manager.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
tensor.cpp Loading commit data...
type_prop.cpp Loading commit data...
update_reference.sh Loading commit data...
util.cpp Loading commit data...
uuid.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...