The added LSTM supports layer norm and projection. Will add LSTM logging kernel for the other variants. PiperOrigin-RevId: 279388926 Change-Id: I84ede1a3994e186a9024eaad344594cfa47199a6
2.5 KiB
2.5 KiB