lemon234071 / Pytorch-TVM-LSTM

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

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.