Dev conv split fw bw (#1759)
* conv * ConvBiasGradOp * fix * fix * conv * conv * ConvFilterGradOp * fix * conv filter kernel * fix * fix * format * fix * fix * fix * fix * conv data grad * conv_bw_data * ConvDataGradKernelUtil * data * conv grad * IsInputBlobAllowedModelSplit * refine * fix * fix * add MakeDataSplitOpParallelSignature * REGISTER_OP * GetCustomizedConf * fix * fix * fix * fix * rename * refine * fix * fix * fix * num_spatial_dims
Showing
- oneflow/core/autograd/conv_grad.cpp 89 additions, 0 deletionsoneflow/core/autograd/conv_grad.cpp
- oneflow/core/device/cudnn_conv_ctx_cache.cpp 130 additions, 21 deletionsoneflow/core/device/cudnn_conv_ctx_cache.cpp
- oneflow/core/device/cudnn_conv_ctx_cache.h 15 additions, 8 deletionsoneflow/core/device/cudnn_conv_ctx_cache.h
- oneflow/core/kernel/conv_bias_grad_kernel.cpp 28 additions, 0 deletionsoneflow/core/kernel/conv_bias_grad_kernel.cpp
- oneflow/core/kernel/conv_bias_grad_kernel.cu 37 additions, 0 deletionsoneflow/core/kernel/conv_bias_grad_kernel.cu
- oneflow/core/kernel/conv_bias_grad_kernel.h 28 additions, 0 deletionsoneflow/core/kernel/conv_bias_grad_kernel.h
- oneflow/core/kernel/conv_data_grad_kernel.cpp 31 additions, 0 deletionsoneflow/core/kernel/conv_data_grad_kernel.cpp
- oneflow/core/kernel/conv_data_grad_kernel.cu 28 additions, 0 deletionsoneflow/core/kernel/conv_data_grad_kernel.cu
- oneflow/core/kernel/conv_data_grad_kernel.h 30 additions, 0 deletionsoneflow/core/kernel/conv_data_grad_kernel.h
- oneflow/core/kernel/conv_filter_grad_kernel.cpp 31 additions, 0 deletionsoneflow/core/kernel/conv_filter_grad_kernel.cpp
- oneflow/core/kernel/conv_filter_grad_kernel.cu 29 additions, 0 deletionsoneflow/core/kernel/conv_filter_grad_kernel.cu
- oneflow/core/kernel/conv_filter_grad_kernel.h 30 additions, 0 deletionsoneflow/core/kernel/conv_filter_grad_kernel.h
- oneflow/core/kernel/kernel.proto 10 additions, 0 deletionsoneflow/core/kernel/kernel.proto
- oneflow/core/operator/conv_bias_grad_op.cpp 106 additions, 0 deletionsoneflow/core/operator/conv_bias_grad_op.cpp
- oneflow/core/operator/conv_bias_grad_op.h 29 additions, 0 deletionsoneflow/core/operator/conv_bias_grad_op.h
- oneflow/core/operator/conv_data_grad_op.cpp 147 additions, 0 deletionsoneflow/core/operator/conv_data_grad_op.cpp
- oneflow/core/operator/conv_data_grad_op.h 29 additions, 0 deletionsoneflow/core/operator/conv_data_grad_op.h
- oneflow/core/operator/conv_filter_grad_op.cpp 166 additions, 0 deletionsoneflow/core/operator/conv_filter_grad_op.cpp
- oneflow/core/operator/conv_filter_grad_op.h 32 additions, 0 deletionsoneflow/core/operator/conv_filter_grad_op.h
- oneflow/core/operator/conv_op.cpp 77 additions, 111 deletionsoneflow/core/operator/conv_op.cpp
Please register or sign in to comment