I need to decompose an LSTM layer into primitive TensorFlow operations. Is it present inside tesnorflow lite model coversion or there are some other methods to do it ?
Please help. Need it for some profiling stuff in ARM processors.
Thanks in advance.
Tensorflow Lite does not provide a specific method for decomposing an LSTM layer into primitive Tensorflow operations.
Please refer to tf.compat.v1.nn.rnn_cell.LSTMCell class to understand how the LSTM cell is implemented in TensorFlow may help you.
Hello @chunduriv ,
Thanks a lot for your reply. This will be helpful.