Tensorflow2 LSTM failed

To convert the model to Relay I had to “fix” begin, end and final_output values inside the code

begin: [(? + -1), 0, 0] -> [0 ,0, 0]
end: [((? + -1) + 1), ?, 512] -> [1, 1, 512]
final_output: [?, 512] -> [1, 512]

here and here