Method build

src/framework/modules/positional_modules.cpp:21–53

    const core::TensorValue & table,
    int64_t half_dim,
    const char * label) {
    core::validate_rank_between(table, 4, 4, label);
    if (!((table.shape.dims[0] == 1 || table.shape.dims[0] == input.shape.dims[0]) &&
          table.shape.dims[1] == input.shape.dims[1] &&
          table.shape.dims[2] == input.shape.dims[2] &&
          table.shape.dims[3] == half_dim)) {
        throw std::runtime_error(std::string("SplitRoPE ") + label + " shape mismatch");
    }
}

core::TensorValue repeat_like(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & value,
    const core::TensorValue & like) {
    if (value.shape.rank == like.shape.rank) {
        bool same = true;
        for (size_t axis = 0; axis < value.shape.rank; ++axis) {
            same = same && value.shape.dims[axis] == like.shape.dims[axis];
        }
        if (same) {
            return value;
        }
    }
    return core::wrap_tensor(ggml_repeat(ctx.ggml, value.tensor, like.tensor), like.shape, value.type);
}

}  // namespace

RoPEModule::RoPEModule(RoPEConfig config) : config_(config) {
    require_positive(config_.dimensions, "RoPEConfig.dimensions");
}

core::TensorValue RoPEModule::build(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const core::TensorValue & positions,