Searched defs:hidden_fn (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2999 inp, hidden, params, has_biases, hidden_fn, batch_sizes, reverse=False argument
3059 def one_layer_rnn(inp, hidden, params, has_biases, hidden_fn, reverse=False): argument
|