Add ReplicationPad2D op (#4190)
* add pad2d ops and kernels * fix bug * add python api and unittest * reformat, change padding_data_type seq, change dim2vector method usage * fix typo * delete ShapeViewToDimVector * rerun make of_format
Showing
- oneflow/python/ops/pad.py 81 additions, 0 deletionsoneflow/python/ops/pad.py
- oneflow/python/test/ops/test_replication_pad2d.py 329 additions, 0 deletionsoneflow/python/test/ops/test_replication_pad2d.py
- oneflow/user/kernels/pad2d_kernels.cpp 250 additions, 0 deletionsoneflow/user/kernels/pad2d_kernels.cpp
- oneflow/user/kernels/pad2d_kernels_util.cpp 92 additions, 0 deletionsoneflow/user/kernels/pad2d_kernels_util.cpp
- oneflow/user/kernels/pad2d_kernels_util.cu 208 additions, 0 deletionsoneflow/user/kernels/pad2d_kernels_util.cu
- oneflow/user/kernels/pad2d_kernels_util.h 247 additions, 0 deletionsoneflow/user/kernels/pad2d_kernels_util.h
- oneflow/user/ops/pad2d_ops.cpp 134 additions, 0 deletionsoneflow/user/ops/pad2d_ops.cpp
- oneflow/user/ops/pad_2d_seq.h 28 additions, 0 deletionsoneflow/user/ops/pad_2d_seq.h
- oneflow/user/ops/reflection_pad2d_op.cpp 0 additions, 131 deletionsoneflow/user/ops/reflection_pad2d_op.cpp
Please register or sign in to comment