This is an automated email from the ASF dual-hosted git repository.
zhaowu pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 6c01998 Make AutoScheduler handling of errors during measure
consistent with AutoTvm (#6909)
add 7e90e7d [Relay] Add space_to_batch_nd and batch_to_space_nd operators
(#6477)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 28 +++
include/tvm/topi/nn.h | 178 +++++++++++++++++
python/tvm/relay/frontend/tensorflow.py | 88 ++-------
python/tvm/relay/frontend/tflite.py | 85 +-------
python/tvm/relay/op/nn/_nn.py | 5 +
python/tvm/relay/op/nn/nn.py | 61 ++++++
python/tvm/relay/op/op_attrs.py | 10 +
python/tvm/topi/nn/__init__.py | 2 +
.../topi/nn/{flatten.py => batch_to_space_nd.py} | 46 ++---
...local_response_norm.py => space_to_batch_nd.py} | 45 ++---
python/tvm/topi/testing/__init__.py | 2 +
python/tvm/topi/testing/batch_to_space_nd.py | 97 +++++++++
python/tvm/topi/testing/space_to_batch_nd.py | 93 +++++++++
src/relay/op/nn/nn.cc | 218 +++++++++++++++++++++
src/topi/nn.cc | 8 +
tests/python/relay/test_op_level5.py | 56 ++++++
.../topi/python/test_topi_batch_to_space_nd.py | 70 +++++++
.../topi/python/test_topi_space_to_batch_nd.py | 72 +++++++
18 files changed, 957 insertions(+), 207 deletions(-)
copy python/tvm/topi/nn/{flatten.py => batch_to_space_nd.py} (51%)
copy python/tvm/topi/nn/{local_response_norm.py => space_to_batch_nd.py} (51%)
create mode 100644 python/tvm/topi/testing/batch_to_space_nd.py
create mode 100644 python/tvm/topi/testing/space_to_batch_nd.py
create mode 100644 tests/python/topi/python/test_topi_batch_to_space_nd.py
create mode 100644 tests/python/topi/python/test_topi_space_to_batch_nd.py
