Support passing LSTM state between lwtnn layers

mmaerker needed to pass the internal state of an LSTM between layers in a Keras model that he's running in Athena. He has a prototype implementation in gitlab but someone needs to integrate this with the standard version.