Hello everyone,
I am looking for a TVM implementation of an optimized multi-layer LSTM for GPU. I tried this one : https://github.com/dmlc/tvm/blob/master/topi/recipe/rnn/lstm.py but it only supports one layer and I don’t know if it is an optimal implementation.
Do you know where I can find this implementation ?
Thank you.