Decomposing LSTM into primitive tf ops

TensorFlow General Discussion
,
1

Hello,
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.

3

@Swaraj_Badhei,

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.

Thank you!

5

Hello @chunduriv ,
Thanks a lot for your reply. This will be helpful.

Cheers,
Swaraj