[Functional] Part6: Migrate conv op (#5252)
* Add partial unary and math functional apis. * Revert elementwise pow. * auto format by CI * Support add with large number of inputs. * Update oneflow/python/nn/modules/math_ops.py Co-authored-by:Yinggang Wang <wyg19970408@gmail.com> * Refine * Migrate binary and activation ops. * Migrate array ops. * Add or refactor activation grad funcs. * Add or refactor activation grad funcs. * Revert unpack all * Fix masked fill * Refine * Add nn ops. * Refine * Refine * Migrate conv op * Fix functional normalization. * auto format by CI * Refine code style * align Torch params * fix bias add error * Support optional parameter. * fix group bug * add new test case * remove useless state * Move optional storage into namespace internal. * add check Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org> Co-authored-by:
Yinggang Wang <wyg19970408@gmail.com> Co-authored-by:
Luyang <flowingsun007@163.com> Co-authored-by:
MARD1NO <359521840@qq.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
ZZK <42901638+MARD1NO@users.noreply.github.com>
Showing
- oneflow/api/python/functional/python_arg.cpp 5 additions, 0 deletionsoneflow/api/python/functional/python_arg.cpp
- oneflow/api/python/functional/python_arg.h 16 additions, 1 deletiononeflow/api/python/functional/python_arg.h
- oneflow/core/autograd/gradient_funcs/conv.cpp 50 additions, 60 deletionsoneflow/core/autograd/gradient_funcs/conv.cpp
- oneflow/core/common/optional.h 202 additions, 0 deletionsoneflow/core/common/optional.h
- oneflow/core/functional/functional_api.yaml 25 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/nn_functor.cpp 40 additions, 0 deletionsoneflow/core/functional/impl/nn_functor.cpp
- oneflow/core/functional/impl/nn_grad_functor.cpp 122 additions, 0 deletionsoneflow/core/functional/impl/nn_grad_functor.cpp
- oneflow/core/functional/value_types.h 6 additions, 0 deletionsoneflow/core/functional/value_types.h
- oneflow/python/nn/modules/conv.py 30 additions, 50 deletionsoneflow/python/nn/modules/conv.py
- oneflow/python/test/modules/test_conv.py 123 additions, 1 deletiononeflow/python/test/modules/test_conv.py
- oneflow/user/kernels/conv_cudnn_kernels.cpp 10 additions, 10 deletionsoneflow/user/kernels/conv_cudnn_kernels.cpp
- oneflow/user/kernels/conv_kernels.cpp 16 additions, 31 deletionsoneflow/user/kernels/conv_kernels.cpp
- tools/generate_functional_api.py 32 additions, 5 deletionstools/generate_functional_api.py
Please register or sign in to comment