Add triu op (#5329)
* add triu
* add triu cuda kernel
* add triu gradient func
* add tensor test for triu
* add triu op
* add hard testcase for triu
* fix format
* add space
* add complicate test case
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/gradient_funcs/triu.cpp 69 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/triu.cpp
- oneflow/core/functional/functional_api.yaml 4 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 14 additions, 0 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/python/nn/modules/triu.py 64 additions, 0 deletionsoneflow/python/nn/modules/triu.py
- oneflow/python/test/modules/test_triu.py 58 additions, 0 deletionsoneflow/python/test/modules/test_triu.py
- oneflow/python/test/tensor/test_tensor.py 37 additions, 0 deletionsoneflow/python/test/tensor/test_tensor.py
- oneflow/user/kernels/triu_kernel.cpp 59 additions, 0 deletionsoneflow/user/kernels/triu_kernel.cpp
- oneflow/user/kernels/triu_kernel.cu 125 additions, 0 deletionsoneflow/user/kernels/triu_kernel.cu
- oneflow/user/ops/triu_op.cpp 50 additions, 0 deletionsoneflow/user/ops/triu_op.cpp
Please register or sign in to comment