Add flip module (#5541)
* fix upsample nearest bug * fix upsample nearest bug (#5347) Co-authored-by:oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> * fix upsample bilinear bug * init flip op * add flip op register * add flip cpu kernel forward * add flip kernel impl * add flip op functor and gradient_funcs * add test, still need fix bug * fix segmentfault bug * add docs * fix comments * fix comments * fix comments Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 2 additions, 0 deletionsdocs/source/experimental.rst
- oneflow/core/autograd/gradient_funcs/flip.cpp 69 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/flip.cpp
- oneflow/core/functional/functional_api.yaml 8 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 32 additions, 0 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/python/nn/modules/flip.py 100 additions, 0 deletionsoneflow/python/nn/modules/flip.py
- oneflow/python/test/modules/test_flip.py 53 additions, 0 deletionsoneflow/python/test/modules/test_flip.py
- oneflow/user/kernels/flip_kernel.cpp 132 additions, 0 deletionsoneflow/user/kernels/flip_kernel.cpp
- oneflow/user/kernels/flip_kernel.cu 132 additions, 0 deletionsoneflow/user/kernels/flip_kernel.cu
- oneflow/user/ops/flip_op.cpp 63 additions, 0 deletionsoneflow/user/ops/flip_op.cpp
Please register or sign in to comment