This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 9a9cd70 [Vulkan][Target] Added the driver name to the vulkan target
string. (#8882)
add 6df070a [ONNX][TOPI] Support select_last_index for argmin/max (#8816)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/reduce.h | 36 +++++++++++
include/tvm/topi/reduction.h | 99 +++++++++++++++++++++++-------
python/tvm/relay/frontend/onnx.py | 20 +++---
python/tvm/relay/op/reduce.py | 20 ++++--
python/tvm/topi/reduction.py | 16 +++--
src/relay/op/tensor/reduce.cc | 80 ++++++++++++++++++------
src/topi/reduction.cc | 4 +-
tests/python/frontend/onnx/test_forward.py | 19 +-----
tests/python/relay/test_op_level4.py | 38 ++++++++++--
9 files changed, 245 insertions(+), 87 deletions(-)
