Posts

Post not yet marked as solved
3 Replies
2.6k Views
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?
Posted Last updated
.
Post not yet marked as solved
0 Replies
884 Views
I have written a tensorflow model that consists of lstm layers - such as tf.contrib.rnn.LSTMBlockCell or tf.keras.layers.LSTM (I can provide the sample code also, if needed). I want to run the model on an iOS app. However, I have looked at several websites that say that presently there is no way to convert and run tensorflow model that consist LSTM layers on iOS apps. I have found these following tools/libraries that may have solve the problem - 1. Swift for Tensorflow2. Tensorflow Lite for iOS3. tfcoremlHowever, these tools also does not seem to be able to convert the lstm layers model to .mlmodel format. These tools, however allow to use custom layers to be added. But I don't know how I can add LSTM custom layer.Am I wrong in saying that there is no support to run tensorflow lstm model in iOS apps? If yes, then please guide me on how I can go ahead to include the model in iOS app. Is there any other tool/library that can be ued to convert it to .mlmodel format. If no, then are there any plans to include tensorflow support in future?
Posted Last updated
.