You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by mb...@apache.org on 2021/12/13 16:00:59 UTC

[tvm] branch main updated (5557b8c -> cb34604)

This is an automated email from the ASF dual-hosted git repository.

mbrookhart pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.


    from 5557b8c  Improve tvmc error message from lazy-loading frontend imports (#9074)
     add cb34604  [TOPI] Add generic batch norm (#9694)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/frontend/onnx.py                |   5 +-
 python/tvm/relay/op/nn/_nn.py                    |   5 +
 python/tvm/relay/op/strategy/generic.py          |  23 +++++
 python/tvm/topi/generic/nn.py                    |  17 ++++
 python/tvm/topi/nn/__init__.py                   |   1 +
 python/tvm/topi/nn/batch_norm.py                 | 110 ++++++++++++++++++++++
 python/tvm/topi/testing/__init__.py              |   1 +
 python/tvm/topi/testing/batch_norm.py            |  89 ++++++++++++++++++
 src/relay/op/nn/nn.cc                            |   3 +-
 src/topi/schedule.cc                             |   3 +
 tests/python/relay/test_op_level1.py             |  48 ++++++++++
 tests/python/topi/python/test_topi_batch_norm.py | 115 +++++++++++++++++++++++
 12 files changed, 418 insertions(+), 2 deletions(-)
 create mode 100644 python/tvm/topi/nn/batch_norm.py
 create mode 100644 python/tvm/topi/testing/batch_norm.py
 create mode 100644 tests/python/topi/python/test_topi_batch_norm.py