Class RMSNormModule

include/engine/framework/modules/norm_modules.h:39–55

        const NormWeights & weights) const;

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

private:
    NormConfig config_;
};

class RMSNormModule {
public:
    explicit RMSNormModule(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;