Function adjacent_repeat_frames_bct

src/models/heartmula/codec.cpp:576–594

    selected = core::ensure_backend_addressable_layout(ctx, modules::TransposeModule({{0, 2, 1}, 3}).build(ctx, selected));
    return core::reshape_tensor(
        ctx,
        selected,
        core::TensorShape::from_dims({codes_bqt.shape.dims[0], codes_bqt.shape.dims[2]}));
}

core::TensorValue conditioning_from_codes(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & codes_bqt,
    const HeartCodecFlowWeights & weights,
    const HeartCodecConfig & config) {
    core::TensorValue summed;
    for (int64_t quantizer = 0; quantizer < config.num_quantizers; ++quantizer) {
        auto indices = reshape_codebook_indices(ctx, codes_bqt, quantizer);
        auto looked_up = modules::CodebookLookupModule({config.codebook_size, config.codebook_dim})
                             .build(ctx, indices, weights.vq_codebooks[static_cast<size_t>(quantizer)]);
        summed = summed.valid() ? modules::AddModule{}.build(ctx, summed, looked_up) : looked_up;
    }
    auto projected = modules::LinearModule({config.codebook_dim, config.dim, true, GGML_PREC_F32})
                         .build(ctx, summed, weights.vq_project_out);
    projected = modules::LinearModule({config.dim, config.dim, true, GGML_PREC_F32})
                    .build(ctx, projected, weights.cond_feature_emb);
    return adjacent_repeat_frames_btc(ctx, projected, 2);