implementation of constantpad-3d op (#5529)
* implementation of constantpad-3d op
* format
* refine
* rename files
* refine
* refine test case
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 4 additions, 3 deletionsdocs/source/experimental.rst
- oneflow/core/autograd/gradient_funcs/padding.cpp 6 additions, 5 deletionsoneflow/core/autograd/gradient_funcs/padding.cpp
- oneflow/core/functional/impl/nn_functor.cpp 10 additions, 1 deletiononeflow/core/functional/impl/nn_functor.cpp
- oneflow/core/functional/impl/nn_grad_functor.cpp 10 additions, 1 deletiononeflow/core/functional/impl/nn_grad_functor.cpp
- oneflow/python/nn/modules/constantpad2d.py 0 additions, 127 deletionsoneflow/python/nn/modules/constantpad2d.py
- oneflow/python/nn/modules/padding.py 170 additions, 0 deletionsoneflow/python/nn/modules/padding.py
- oneflow/python/test/modules/test_constantpad.py 26 additions, 3 deletionsoneflow/python/test/modules/test_constantpad.py
- oneflow/user/kernels/constantpad3d_kernel.cpp 147 additions, 0 deletionsoneflow/user/kernels/constantpad3d_kernel.cpp
- oneflow/user/kernels/constantpad3d_kernel_util.cpp 64 additions, 0 deletionsoneflow/user/kernels/constantpad3d_kernel_util.cpp
- oneflow/user/kernels/constantpad3d_kernel_util.cu 125 additions, 0 deletionsoneflow/user/kernels/constantpad3d_kernel_util.cu
- oneflow/user/kernels/constantpad3d_kernel_util.h 122 additions, 0 deletionsoneflow/user/kernels/constantpad3d_kernel_util.h
- oneflow/user/ops/padding_ops.cpp 109 additions, 0 deletionsoneflow/user/ops/padding_ops.cpp
Please register or sign in to comment