Class LayerNormModule

include/engine/framework/modules/norm_modules.h:21–37

    std::optional<core::TensorValue> bias;
};

struct ChannelAffineWeights {
    core::TensorValue scale;
    core::TensorValue bias;
};

class LayerNormModule {
public:
    explicit LayerNormModule(NormConfig config);

    const core::ModuleSchema & schema() const noexcept;
    const NormConfig & config() const noexcept;

    core::TensorValue build(
        core::ModuleBuildContext & ctx,
        const core::TensorValue & input,
        const NormWeights & weights) const;

    static const core::ModuleSchema & static_schema() noexcept;