Add gradient funcs. (#5348)
Co-authored-by:Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/gradient_funcs/add_n.cpp 54 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/add_n.cpp
- oneflow/core/autograd/gradient_funcs/clip_by_scalar.cpp 75 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/clip_by_scalar.cpp
- oneflow/core/autograd/gradient_funcs/clip_by_scalar_min.cpp 72 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/clip_by_scalar_min.cpp
- oneflow/core/autograd/gradient_funcs/gelu.cpp 54 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/gelu.cpp
- oneflow/core/autograd/gradient_funcs/hardsigmoid.cpp 54 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/hardsigmoid.cpp
- oneflow/core/autograd/gradient_funcs/hardswish.cpp 54 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/hardswish.cpp
- oneflow/core/autograd/gradient_funcs/multiply.cpp 76 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/multiply.cpp
- oneflow/core/autograd/gradient_funcs/relu.cpp 54 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/relu.cpp
- oneflow/core/functional/functional_api.yaml 40 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/activation_functor.cpp 33 additions, 0 deletionsoneflow/core/functional/impl/activation_functor.cpp
- oneflow/core/functional/impl/math_functor.cpp 160 additions, 0 deletionsoneflow/core/functional/impl/math_functor.cpp
Please register or sign in to comment