lemon234071 / Pytorch-TVM-LSTM Public

Notifications You must be signed in to change notification settings
Fork 1
Star 2

A implement LSTM of pytorch version compiled by TVM, backend is LLVM.

2 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
datasets/images		datasets/images
.gitignore		.gitignore
README.md		README.md
compile_demo.py		compile_demo.py
lstm-py.py		lstm-py.py
lstm.onnx		lstm.onnx
test.py		test.py

Repository files navigation

Pytorch-TVM-LSTM

A implement LSTM of pytorch version compiled by TVM, backend is LLVM.

lstm-py is a simple implement of LSTM by pytorch.

Dataset is the mnist, and the task is a classification.

test.py test the relay form onnx model.

compile_demo is a demo of compile onnx model.

TODO implement op of pytorch LSTM.

About

A implement LSTM of pytorch version compiled by TVM, backend is LLVM.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%