Enzyme main
Loading...
Searching...
No Matches
mlir::impulse::WelfordState Struct Reference

State for Welford covariance estimation. More...

#include "MLIR/Interfaces/HMCUtils.h"

Collaboration diagram for mlir::impulse::WelfordState:

Public Member Functions

SmallVector< Value > toValues () const
 
SmallVector< Type > getTypes () const
 

Static Public Member Functions

static WelfordState fromValues (ArrayRef< Value > values)
 

Public Attributes

Value mean
 
Value m2
 
Value n
 

Detailed Description

State for Welford covariance estimation.

Definition at line 405 of file HMCUtils.h.

Member Function Documentation

◆ fromValues()

static WelfordState mlir::impulse::WelfordState::fromValues ( ArrayRef< Value > values)
inlinestatic

Definition at line 411 of file HMCUtils.h.

◆ getTypes()

SmallVector< Type > mlir::impulse::WelfordState::getTypes ( ) const
inline

Definition at line 414 of file HMCUtils.h.

References m2, mean, and n.

◆ toValues()

SmallVector< Value > mlir::impulse::WelfordState::toValues ( ) const
inline

Definition at line 410 of file HMCUtils.h.

References m2, mean, and n.

Member Data Documentation

◆ m2

Value mlir::impulse::WelfordState::m2

◆ mean

Value mlir::impulse::WelfordState::mean

Definition at line 406 of file HMCUtils.h.

Referenced by getTypes(), toValues(), and mlir::impulse::updateWelford().

◆ n

Value mlir::impulse::WelfordState::n

The documentation for this struct was generated from the following file: