Method build

src/framework/modules/conditioning_modules.cpp:215–221

}  // namespace

core::TensorValue LayerScaleModule::build(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const LayerScaleWeights & weights) const {
    auto scale = repeat_like(ctx, weights.scale, input);
    return mul(ctx, input, scale);
}

TimestepEmbeddingModule::TimestepEmbeddingModule(TimestepEmbeddingConfig config) : config_(config) {