Lines Matching refs:prev_state
36 def vanilla_lstm(num_hidden, indata, prev_state, param, seqidx, layeridx, is_batchnorm=False, gamma… argument
48 h2h = mx.sym.FullyConnected(data=prev_state.h,
60 next_c = (forget_gate * prev_state.c) + (in_gate * in_transform)
65 def lstm(num_hidden, indata, prev_state, param, seqidx, layeridx, dropout=0., num_hidden_proj=0, is… argument
83 h2h = mx.sym.FullyConnected(data=prev_state.h,
94 Wcidc = mx.sym.broadcast_mul(param.c2i_bias, prev_state.c) + slice_gates[0]
99 Wcfdc = mx.sym.broadcast_mul(param.c2f_bias, prev_state.c) + slice_gates[2]
102 next_c = (forget_gate * prev_state.c) + (in_gate * in_transform)
179 prev_state=last_states[i],
193 prev_state=last_states[i],
205 prev_state=last_states[i],
215 prev_state=last_states[i],