Add conv3d Module (#5327)
* add conv3d module * add simple test case * use conv base class to write conv3d * still test error * add torch style conv3d unit test * fix format * add assert * unittest still error * auto format by CI * fix format and autotest * remove dir * remove useless file * add extra expr * auto format by CI * fix import * fix doc Co-authored-by:oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- oneflow/core/functional/functional_api.yaml 6 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/nn_functor.cpp 9 additions, 0 deletionsoneflow/core/functional/impl/nn_functor.cpp
- python/oneflow/nn/__init__.py 1 addition, 1 deletionpython/oneflow/nn/__init__.py
- python/oneflow/nn/modules/conv.py 206 additions, 2 deletionspython/oneflow/nn/modules/conv.py
- python/oneflow/test/modules/test_conv3d.py 46 additions, 0 deletionspython/oneflow/test/modules/test_conv3d.py
Please register or sign in to comment