Add pad grad func. (#5354)
* Add pad grad func.
* Fix constant pad grad func.
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/gradient_funcs/pad2d.cpp 133 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/pad2d.cpp
- oneflow/core/functional/functional_api.yaml 18 additions, 4 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/nn_functor.cpp 45 additions, 3 deletionsoneflow/core/functional/impl/nn_functor.cpp
- oneflow/core/functional/impl/nn_grad_functor.cpp 45 additions, 0 deletionsoneflow/core/functional/impl/nn_grad_functor.cpp
- oneflow/core/functional/scalar.h 16 additions, 6 deletionsoneflow/core/functional/scalar.h
- oneflow/python/nn/modules/padding.py 2 additions, 18 deletionsoneflow/python/nn/modules/padding.py
Please register or sign in to comment