Optimize reduce mean (#5452)
* Optimize reduce mean
* Fix mean axis.
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/functional/functional_api.yaml 4 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/math_functor.cpp 27 additions, 1 deletiononeflow/core/functional/impl/math_functor.cpp
- oneflow/python/nn/modules/activation.py 1 addition, 2 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/nn/modules/math_ops.py 1 addition, 11 deletionsoneflow/python/nn/modules/math_ops.py
- oneflow/python/nn/modules/reduce_ops.py 2 additions, 15 deletionsoneflow/python/nn/modules/reduce_ops.py
Please register or sign in to comment