gussmith23 / glenside

A pure, low-level tensor program representation enabling tensor program optimization via program rewriting. See the web demo at https://gussmith23.github.io/glenside-web-demo/
68 stars 10 forks source link

Support LSTM #138

Closed gussmith23 closed 2 years ago

gussmith23 commented 2 years ago

Adds support for the LSTM which we would like to match on in the 3LA PLDI push. Currently we are doing "static matching" using TVM and a pre-set pattern. We would like to also support "flexible matching", meaning we need to bring the LSTM in to Glenside.

gussmith23 commented 2 years ago

Closing in favor of #139