Support Relay's `nn.batch_matmul`

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/

71 stars 10 forks source link

Support Relay's `nn.batch_matmul` #141

Open gussmith23 opened 2 years ago

gussmith23 commented 2 years ago

This is a good example of a more complex operator that should still definitely be representable in Glenside. It appears in Transformer.