Dev minimum maximum (#5576)
* add minimum and maximum
* add testcase
* fix docs
* move to python
* add elementwise min max grad func
* add autotest
* add broadcast min max grad func
* add broadcast min max testcase
* add broadcast_binary related functional
* convert gradient func to functional
* delete elementwise op_expr
* delete grad_op
* bind python
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/gradient_funcs/broadcast_binary_ops.cpp 92 additions, 104 deletions...low/core/autograd/gradient_funcs/broadcast_binary_ops.cpp
- oneflow/core/autograd/gradient_funcs/elementwise_minimum_maximum.cpp 83 additions, 0 deletions...e/autograd/gradient_funcs/elementwise_minimum_maximum.cpp
- oneflow/core/framework/op_expr_helper.cpp 1 addition, 0 deletionsoneflow/core/framework/op_expr_helper.cpp
- oneflow/core/functional/functional_api.yaml 40 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 98 additions, 0 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/core/functional/impl/binary_functor.cpp 43 additions, 0 deletionsoneflow/core/functional/impl/binary_functor.cpp
- python/oneflow/__init__.py 2 additions, 0 deletionspython/oneflow/__init__.py
- python/oneflow/nn/modules/math_ops.py 86 additions, 0 deletionspython/oneflow/nn/modules/math_ops.py
- python/oneflow/test/modules/test_math_ops.py 155 additions, 0 deletionspython/oneflow/test/modules/test_math_ops.py
Please register or sign in to comment