I have a simple tensorflow model with lstm layers. I want to convert the model to .mlmodel format. However, I think, as of now, CoreML does not support LSTM layers and hence I need to write a custom lstm layer in swift.
How can I write that custom layer?
Convert Tensorflow LSTM model to CoreML
Add a Comment