Finding TensorFlow equivalent of Pytorch GRU feature

Question

I am confused about how to reconstruct the following Pytorch code in TensorFlow. It uses both the input size x and the hidden size h to create a GR