MCPcopy Create free account
hub / github.com/pytorch/pytorch / GetLSTMParamNames

Function GetLSTMParamNames

caffe2/python/rnn_cell.py:1665–1668  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1663
1664
1665def GetLSTMParamNames():
1666 weight_params = ["input_gate_w", "forget_gate_w", "output_gate_w", "cell_w"]
1667 bias_params = ["input_gate_b", "forget_gate_b", "output_gate_b", "cell_b"]
1668 return {'weights': weight_params, 'biases': bias_params}
1669
1670
1671def InitFromLSTMParams(lstm_pblobs, param_values):

Callers 2

InitFromLSTMParamsFunction · 0.85
cudnn_LSTMFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…