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 6c04ac52bd [TRT][BYOC] allow strided_slice ops on selected dimensions
(#14142) (#14144)
add 69acdfb042 [ONNX][TOPI] Add `DFT` operator (#13999)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/transform.h | 11 +++
python/tvm/relay/frontend/onnx.py | 111 ++++++++++++++++++++++++++++
python/tvm/relay/op/_transform.py | 14 ++++
python/tvm/relay/op/strategy/cuda.py | 11 +++
python/tvm/relay/op/strategy/generic.py | 26 +++++++
python/tvm/relay/op/transform.py | 27 +++++++
python/tvm/topi/__init__.py | 2 +-
python/tvm/topi/cuda/__init__.py | 2 +-
python/tvm/topi/cuda/{stft.py => signal.py} | 96 ++++++++++++++++++++++++
python/tvm/topi/{stft.py => signal.py} | 82 ++++++++++++++++++++
src/relay/op/tensor/transform.cc | 49 ++++++++++++
tests/python/frontend/onnx/test_forward.py | 78 ++++++++++++++++++-
tests/python/topi/python/test_topi_dft.py | 88 ++++++++++++++++++++++
13 files changed, 592 insertions(+), 5 deletions(-)
rename python/tvm/topi/cuda/{stft.py => signal.py} (60%)
rename python/tvm/topi/{stft.py => signal.py} (62%)
create mode 100644 tests/python/topi/python/test_topi_dft.py