Support inplace add (#5432)
* feat(add_): support inplace add
* fix(add_): fix push bug
* Add InplaceableBinaryFunctor
* feat(add_): add tensor += other
* docs(add_): fix docs
* fix relu module bug
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- docs/source/experimental.rst 1 addition, 0 deletionsdocs/source/experimental.rst
- oneflow/core/autograd/autograd_engine.cpp 2 additions, 1 deletiononeflow/core/autograd/autograd_engine.cpp
- oneflow/core/framework/tensor_arg.cpp 1 addition, 1 deletiononeflow/core/framework/tensor_arg.cpp
- oneflow/core/functional/functional_api.yaml 5 additions, 5 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/binary_functor.cpp 3 additions, 3 deletionsoneflow/core/functional/impl/binary_functor.cpp
- oneflow/core/functional/impl/binary_functor.h 21 additions, 0 deletionsoneflow/core/functional/impl/binary_functor.h
- oneflow/core/functional/impl/math_functor.cpp 21 additions, 6 deletionsoneflow/core/functional/impl/math_functor.cpp
- oneflow/python/framework/tensor.py 4 additions, 0 deletionsoneflow/python/framework/tensor.py
- oneflow/python/nn/modules/activation.py 5 additions, 8 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/nn/modules/math_ops.py 42 additions, 9 deletionsoneflow/python/nn/modules/math_ops.py
- oneflow/python/nn/modules/utils.py 7 additions, 0 deletionsoneflow/python/nn/modules/utils.py
- oneflow/python/test/modules/test_add.py 93 additions, 2 deletionsoneflow/python/test/modules/test_add.py
Please register or sign in to comment