Skip to content

Latest commit

 

History

History
8 lines (7 loc) · 333 Bytes

README.md

File metadata and controls

8 lines (7 loc) · 333 Bytes

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.