Is there a way to mark a layer as an LSTM layer after conversion to relay? I see almost RNN ops as broken into individual cells for each operator. I have a custom accelerator with First class support for LSTM at the operator level. Is there an easy way to do this ?
Thanks in advance.