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 4d36a45b9f [Relax][ONNX] Add image.resize3d op and wire 5D Resize
(#18931)
add 1e08eb2fa9 [Relax][ONNX][Torch] Add roi_align support and frontend
integration (#18936)
add bf6ed31302 [Relax][PyTorch] Add 3D interpolate support using resize3d
(#18937)
add 2f2469e637 [Relax] Add affine_grid operator with PyTorch and ONNX
frontend support (#18933)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/vision.h | 25 +++
python/tvm/relax/frontend/onnx/onnx_frontend.py | 101 +++++++++-
.../frontend/torch/exported_program_translator.py | 61 ++++++
python/tvm/relax/frontend/torch/fx_translator.py | 40 +++-
python/tvm/relax/op/__init__.py | 2 +-
python/tvm/relax/op/image/__init__.py | 2 +-
python/tvm/relax/op/image/image.py | 41 ++++
python/tvm/relax/op/op_attrs.py | 5 +
python/tvm/relax/op/vision/__init__.py | 1 +
python/tvm/relax/op/vision/roi_align.py | 78 ++++++++
python/tvm/relax/transform/legalize_ops/image.py | 18 +-
python/tvm/relax/transform/legalize_ops/vision.py | 15 ++
python/tvm/topi/testing/roi_align_python.py | 15 +-
python/tvm/topi/vision/__init__.py | 1 +
python/tvm/topi/vision/roi_align.py | 204 +++++++++++++++++++
src/relax/op/image/resize.cc | 94 +++++++++
src/relax/op/image/resize.h | 3 +
src/relax/op/vision/roi_align.cc | 141 +++++++++++++
.../op/{nn/attention.h => vision/roi_align.h} | 19 +-
.../relax/test_frontend_from_exported_program.py | 117 +++++++++++
tests/python/relax/test_frontend_from_fx.py | 222 +++++++++++++++++++++
tests/python/relax/test_frontend_onnx.py | 86 +++++++-
tests/python/relax/test_op_image.py | 131 ++++++++++++
tests/python/relax/test_op_vision.py | 168 ++++++++++++++++
.../relax/test_transform_legalize_ops_image.py | 36 ++++
.../relax/test_tvmscript_parser_op_vision.py | 32 +++
26 files changed, 1626 insertions(+), 32 deletions(-)
create mode 100644 python/tvm/relax/op/vision/roi_align.py
create mode 100644 python/tvm/topi/vision/roi_align.py
create mode 100644 src/relax/op/vision/roi_align.cc
copy src/relax/op/{nn/attention.h => vision/roi_align.h} (65%)
