tenstorrent / tt-mlir

Tenstorrent MLIR compiler
https://tenstorrent.github.io/tt-mlir/
Apache License 2.0
76 stars 13 forks source link

MNIST op modeling - softmax (PoC) #1354

Open mbezuljTT opened 1 day ago

mbezuljTT commented 1 day ago

Create op model for TTNN softmax op, at minimum it needs to cover MNIST tensor shapes, ideally we want to cover to full L1.