You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2022/06/07 19:21:54 UTC
[tvm] branch last-successful updated (70884e957a -> 32a86f8304)
This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 70884e957a fix uint case (#11597)
add 32a86f8304 [TOPI] TE implementation of LSTM using scan (#11531)
No new revisions were added by this update.
Summary of changes:
python/tvm/topi/generic/nn.py | 16 ++
python/tvm/topi/nn/__init__.py | 1 +
python/tvm/topi/nn/lstm.py | 235 +++++++++++++++++++++++++++++
python/tvm/topi/testing/__init__.py | 1 +
python/tvm/topi/testing/lstm_python.py | 134 ++++++++++++++++
tests/python/topi/python/test_topi_lstm.py | 161 ++++++++++++++++++++
6 files changed, 548 insertions(+)
create mode 100644 python/tvm/topi/nn/lstm.py
create mode 100644 python/tvm/topi/testing/lstm_python.py
create mode 100644 tests/python/topi/python/test_topi_lstm.py