Searched defs:ih_weight (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 3045 def inner(i, cur_hidden, ih_weight, ih_bias, hh_weight, hh_bias): argument
3052 def inner(i, cur_hidden, ih_weight, ih_bias, hh_weight, hh_bias): argument
3525 def gru_cell(inp, cur_hidden, ih_weight, ih_bias, hh_weight, hh_bias): argument
3534 def gru_cell_data(inp, cur_hidden, ih_weight, ih_bias, hh_weight, hh_bias): argument
|