73844a5b82ca5a3dd9837a3fbed434b06af360bd,thinc/layers/lstm.py,,forward,#Any#Any#Any#,83

Before Change

def forward(
    model: Model[RNNState, RNNState], prevstate_inputs: RNNState, is_train: bool
) -> Tuple[RNNState, Callable]:
    (cell_tm1, hidden_tm1), inputs = prevstate_inputs
    weights = model.layers[0]
    W = weights.get_param("W")
    b = weights.get_param("b")

After Change

    W = model.get_param("W")
    b = model.get_param("b")
    h = model.get_param("h")
    c = model.get_param("c")
    // Initialize hiddens and cells
    hiddens = model.ops.alloc_f2d(X.shape[1], h.shape[0])
    cells = model.ops.alloc_f2d(X.shape[1], c.shape[0])
    hiddens += h

Instances