How to modify the internal calculation process of LSTM in pytorch-v1.1.0?

I want to modify the calculation process inside the LSTM. However, when I queried the _VF.lstm() method, no corresponding python implementation was found. Then I found the C++ implementation at this address (i.e., https://github.com/pytorch/pytorch/tree/master/aten/src/ATen/native/RNN.cpp) on GitHub. My question is which files need to be modified under the local PyTorch directory.