Fix upsample bilinear bug (#5366)
* 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 * fix upsample bilinear bug * recover code * align with pytorch * redesign upsample bilinear * fix align corner bug * align bilinear with pytorch * fix conflict * fix bug * code reuse * fix commets Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/test/modules/test_upsample2d.py 32 additions, 4 deletionsoneflow/python/test/modules/test_upsample2d.py
- oneflow/user/kernels/upsample_kernel.cpp 1 addition, 57 deletionsoneflow/user/kernels/upsample_kernel.cpp
- oneflow/user/kernels/upsample_kernel.cu 1 addition, 56 deletionsoneflow/user/kernels/upsample_kernel.cu
- oneflow/user/kernels/upsample_kernel.h 80 additions, 0 deletionsoneflow/user/kernels/upsample_kernel.h
Please register or sign in to comment