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 d043cb9 [BugFix][TOPI] Fix the integer overflow problem of the
scatter_nd op. (#8415)
add 807373c Add qnn batch_matmul operator (#8401)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/strategy/x86.py | 6 +-
python/tvm/relay/qnn/op/qnn.py | 38 ++++
python/tvm/topi/nn/batch_matmul.py | 28 ++-
python/tvm/topi/x86/batch_matmul.py | 25 ++-
src/relay/op/nn/nn.cc | 53 +-----
src/relay/op/nn/nn.h | 54 ++++++
src/relay/qnn/op/batch_matmul.cc | 216 +++++++++++++++++++++
tests/python/relay/test_op_qnn_batch_matmul.py | 247 +++++++++++++++++++++++++
8 files changed, 600 insertions(+), 67 deletions(-)
create mode 100644 src/relay/qnn/op/batch_matmul.cc
create mode 100644 tests/python/relay/test_op_qnn_batch_matmul.py
