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

Reply via email to