Method build

src/framework/modules/recurrent_modules.cpp:218–239

}

LSTMCellOutputs LSTMCellModule::build(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const core::TensorValue & hidden,
    const core::TensorValue & cell,
    const LSTMCellWeights & weights) const {
    if (ctx.ggml == nullptr) {
        throw std::runtime_error("ModuleBuildContext.ggml is null");
    }
    core::validate_rank_between(input, 2, 2, "input");
    core::validate_shape(hidden, core::TensorShape::from_dims({input.shape.dims[0], config_.hidden_size}), "hidden");
    core::validate_shape(cell, core::TensorShape::from_dims({input.shape.dims[0], config_.hidden_size}), "cell");
    core::validate_shape(input, core::TensorShape::from_dims({input.shape.dims[0], config_.input_size}), "input");
    validate_weight_shapes(config_, weights);

    const int64_t gates = 4 * config_.hidden_size;
    const auto projected_input = LinearModule({config_.input_size, gates, true}).build(
        ctx,
        input,
        {weights.weight_ih, weights.bias_ih});
    return build_lstm_cell_from_projected_input(ctx, config_, projected_input, hidden, cell, weights);
}

const core::ModuleSchema & LSTMCellModule::static_schema() noexcept {
    return kLstmSchema;