This is an automated email from the ASF dual-hosted git repository.
yongwww pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 601d570894 [CI] Upgrade ubuntu runner image for GitHub CI (#17846)
add 103e54bf4f [Relax][PyTorch] Add PReLU Op Support for Exported Program
and FX graph (#17816)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/nn.h | 9 +++++
python/tvm/relax/frontend/nn/op.py | 28 ++++++++++++++++
.../frontend/torch/base_fx_graph_translator.py | 6 ++++
.../frontend/torch/exported_program_translator.py | 1 +
python/tvm/relax/frontend/torch/fx_translator.py | 10 ++++++
python/tvm/relax/op/nn/__init__.py | 1 +
python/tvm/relax/op/nn/nn.py | 26 +++++++++++++++
python/tvm/relax/transform/legalize_ops/nn.py | 5 +++
python/tvm/topi/nn/elemwise.py | 3 ++
src/relax/op/nn/nn.cc | 21 ++++++++++++
src/relax/op/nn/nn.h | 3 ++
.../relax/test_frontend_from_exported_program.py | 36 ++++++++++++++++++++
tests/python/relax/test_frontend_from_fx.py | 39 ++++++++++++++++++++++
tests/python/relax/test_frontend_nn_op.py | 2 ++
tests/python/relax/test_op_nn.py | 4 +++
15 files changed, 194 insertions(+)
