This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 52b5d55fc4 [Frontend][ONNX] Add If operator support to Relax ONNX
frontend (#18946)
add 4de1f11344 [Relax] Add conv3d_transpose and ONNX ConvTranspose 3D
support (#18948)
add 4df6b1750b [Relax][ONNX] Add roi_pool op and MaxRoiPool frontend
support (#18952)
add 8597d21a8d [Frontend][ONNX] Add MatMulInteger support to Relax ONNX
frontend (#18951)
add e229bda76f [Docs] Add tutorial for mixing Python/PyTorch with TVM
using BasePyModule (#18947)
No new revisions were added by this update.
Summary of changes:
.../tutorials/mix_python_and_tvm_with_pymodule.py | 468 +++++++++++++++++++++
include/tvm/relax/attrs/nn.h | 52 +++
include/tvm/relax/attrs/vision.h | 18 +-
python/tvm/relax/base_py_module.py | 9 +-
python/tvm/relax/frontend/onnx/onnx_frontend.py | 84 +++-
python/tvm/relax/op/__init__.py | 1 +
python/tvm/relax/op/nn/__init__.py | 1 +
python/tvm/relax/op/nn/nn.py | 106 +++++
python/tvm/relax/op/op_attrs.py | 10 +
python/tvm/relax/op/vision/__init__.py | 1 +
.../relax/op/vision/{roi_align.py => roi_pool.py} | 31 +-
python/tvm/relax/transform/legalize_ops/nn.py | 38 +-
python/tvm/relax/transform/legalize_ops/vision.py | 12 +
python/tvm/runtime/support.py | 11 +-
python/tvm/s_tir/meta_schedule/utils.py | 11 +-
python/tvm/topi/vision/__init__.py | 1 +
python/tvm/topi/vision/roi_pool.py | 94 +++++
src/relax/op/nn/convolution.cc | 247 ++++++++++-
src/relax/op/nn/convolution.h | 12 +
src/relax/op/vision/{roi_align.cc => roi_pool.cc} | 63 ++-
src/relax/op/vision/{roi_align.h => roi_pool.h} | 16 +-
tests/python/relax/test_frontend_onnx.py | 225 +++++++++-
tests/python/relax/test_op_nn_convolution.py | 74 ++++
tests/python/relax/test_op_vision.py | 99 ++++-
.../python/relax/test_transform_legalize_ops_nn.py | 108 +++++
tests/python/relax/test_tvmscript_parser_op_nn.py | 38 ++
26 files changed, 1730 insertions(+), 100 deletions(-)
create mode 100644 docs/how_to/tutorials/mix_python_and_tvm_with_pymodule.py
copy python/tvm/relax/op/vision/{roi_align.py => roi_pool.py} (68%)
create mode 100644 python/tvm/topi/vision/roi_pool.py
copy src/relax/op/vision/{roi_align.cc => roi_pool.cc} (61%)
copy src/relax/op/vision/{roi_align.h => roi_pool.h} (69%)
