You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by wu...@apache.org on 2020/06/17 20:39:20 UTC
[incubator-tvm] branch master updated (3463528 -> 052ea4d)
This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 3463528 [Frontend][MXNet] Support a few contrib ops in mxnet (#5819)
add 052ea4d Add a combine batch_matmul pass (#5791)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/transform.h | 11 ++
python/tvm/relay/transform/transform.py | 34 +++++
src/relay/backend/build_module.cc | 1 +
.../transforms/combine_parallel_batch_matmul.cc | 160 +++++++++++++++++++++
.../test_pass_combine_parallel_batch_matmul.py | 146 +++++++++++++++++++
5 files changed, 352 insertions(+)
create mode 100644 src/relay/transforms/combine_parallel_batch_matmul.cc
create mode 100644 tests/python/relay/test_pass_combine_parallel_batch_matmul.py