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

Posted to commits@tvm.apache.org by co...@apache.org on 2021/02/26 17:31:49 UTC

[tvm] branch main updated (c46b187 -> 56ac892)

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

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


    from c46b187  [Torch] Pool ops, convert strides and pool_size to int (#7517)
     add 56ac892  SparseReshape Op (#7477)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/frontend/tensorflow.py          |  10 +
 python/tvm/relay/op/_transform.py                |  35 ++++
 python/tvm/relay/op/strategy/cuda.py             |  11 ++
 python/tvm/relay/op/strategy/generic.py          |  27 +++
 python/tvm/relay/op/transform.py                 |  40 ++++
 python/tvm/topi/__init__.py                      |   1 +
 python/tvm/topi/cuda/__init__.py                 |   1 +
 python/tvm/topi/cuda/sparse_reshape.py           | 209 +++++++++++++++++++++
 python/tvm/topi/sparse_reshape.py                | 185 +++++++++++++++++++
 src/relay/op/tensor/transform.cc                 |  50 +++++
 tests/python/frontend/tensorflow/test_forward.py | 124 +++++++++++++
 tests/python/relay/test_op_level3.py             | 223 +++++++++++++++++++++++
 12 files changed, 916 insertions(+)
 create mode 100644 python/tvm/topi/cuda/sparse_reshape.py
 create mode 100644 python/tvm/topi/sparse_reshape.py