add greater_less_argmax module (#4756)
* add greater_less_argmax module * fix comment * add test_case * fix conflict * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * fix comment * format file Co-authored-by:daquexian <daquexian566@gmail.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/modules/argmax.py 104 additions, 0 deletionsoneflow/python/nn/modules/argmax.py
- oneflow/python/nn/modules/greater.py 65 additions, 0 deletionsoneflow/python/nn/modules/greater.py
- oneflow/python/nn/modules/less.py 62 additions, 0 deletionsoneflow/python/nn/modules/less.py
- oneflow/python/nn/modules/squeeze.py 71 additions, 0 deletionsoneflow/python/nn/modules/squeeze.py
- oneflow/python/nn/modules/transpose.py 76 additions, 0 deletionsoneflow/python/nn/modules/transpose.py
- oneflow/python/test/modules/test_argmax.py 67 additions, 0 deletionsoneflow/python/test/modules/test_argmax.py
- oneflow/python/test/modules/test_greater.py 43 additions, 0 deletionsoneflow/python/test/modules/test_greater.py
- oneflow/python/test/modules/test_less.py 43 additions, 0 deletionsoneflow/python/test/modules/test_less.py
- oneflow/python/test/modules/test_squeeze.py 41 additions, 0 deletionsoneflow/python/test/modules/test_squeeze.py
- oneflow/python/test/modules/test_transpose.py 41 additions, 0 deletionsoneflow/python/test/modules/test_transpose.py
oneflow/python/nn/modules/argmax.py
0 → 100644
oneflow/python/nn/modules/greater.py
0 → 100644
oneflow/python/nn/modules/less.py
0 → 100644
oneflow/python/nn/modules/squeeze.py
0 → 100644
oneflow/python/nn/modules/transpose.py
0 → 100644
oneflow/python/test/modules/test_argmax.py
0 → 100644
oneflow/python/test/modules/test_greater.py
0 → 100644
oneflow/python/test/modules/test_less.py
0 → 100644
oneflow/python/test/modules/test_squeeze.py
0 → 100644
Please register or sign in to comment