Method build

src/framework/modules/predictor_modules.cpp:13–25

DurationPredictorModule::DurationPredictorModule(DurationPredictorConfig config) : config_(config) {}

core::TensorValue DurationPredictorModule::build(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    const DurationPredictorWeights & weights) const {
    auto x = tensor_layout::swap_channel_time_axes_3d(ctx, input);
    x = Conv1dModule({config_.hidden_size, config_.channels, 3, 1, 1, 1, config_.use_bias}).build(ctx, x, weights.conv1);
    x = ReluModule().build(ctx, x);
    x = Conv1dModule({config_.channels, config_.channels, 3, 1, 1, 1, config_.use_bias}).build(ctx, x, weights.conv2);
    x = ReluModule().build(ctx, x);
    x = Conv1dModule({config_.channels, 1, 1, 1, 0, 1, config_.use_bias}).build(ctx, x, weights.proj);
    x = core::wrap_tensor(ggml_cont(ctx.ggml, x.tensor), x.shape, x.type);
    return core::reshape_tensor(ctx, x, core::TensorShape::from_dims({input.shape.dims[0], input.shape.dims[1]}));
}

VariancePredictorModule::VariancePredictorModule(DurationPredictorConfig config) : config_(config) {}