Function adaln_embedding

src/models/heartmula/codec.cpp:746–757

            ctx,
            core::ensure_backend_addressable_layout(ctx, timestep_mod),
            core::TensorShape::from_dims({batch, 6, hidden_size})),
        scale_shift);
    AdaNormParts out{
        expand_batch_token(ctx, modules::SliceModule({1, 0, 1}).build(ctx, parts), tokens, hidden_size),
        expand_batch_token(ctx, modules::SliceModule({1, 1, 1}).build(ctx, parts), tokens, hidden_size),
        expand_batch_token(ctx, modules::SliceModule({1, 2, 1}).build(ctx, parts), tokens, hidden_size),
        expand_batch_token(ctx, modules::SliceModule({1, 3, 1}).build(ctx, parts), tokens, hidden_size),
        expand_batch_token(ctx, modules::SliceModule({1, 4, 1}).build(ctx, parts), tokens, hidden_size),
        expand_batch_token(ctx, modules::SliceModule({1, 5, 1}).build(ctx, parts), tokens, hidden_size),
    };
    return out;
}

core::TensorValue flow_attention(
    core::ModuleBuildContext & ctx,