Bump because I’m looking for a similar thing – can’t find the definition of ‘torch._C._VariableFunctions.gru’ anywhere in the source code.
All I know is that nn.GRU implements a different forward computation from that defined in one of the gru_cell 9 functions, because with exactly the same parameters they give substantially different activities.
Found the same links and c++ files as OP, but no luck either. All I wanted was to access my GRU update and reset gate activations …