About LSTM models

haichen · November 8, 2019, 5:27pm

MxNet test cases cover the LSTM model.

apache/incubator-tvm/blob/master/tests/python/frontend/mxnet/test_forward.py#L598-L603


for mode in ["rnn", "gru", "lstm"]:
    verify(mode, 1, 64, 64, 1)
    verify(mode, 10, 64, 64, 2)
    verify(mode, 10, 64, 32, 2)
    verify(mode, 10, 64, 32, 2, batch=2)
    verify(mode, 10, 32, 64, 1, bidirectional=True)