|
Enzyme main
|
State for Welford covariance estimation. More...
#include "MLIR/Interfaces/HMCUtils.h"

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 |
State for Welford covariance estimation.
Definition at line 405 of file HMCUtils.h.
|
inlinestatic |
Definition at line 411 of file HMCUtils.h.
|
inline |
Definition at line 414 of file HMCUtils.h.
|
inline |
Definition at line 410 of file HMCUtils.h.
| Value mlir::impulse::WelfordState::m2 |
Definition at line 407 of file HMCUtils.h.
Referenced by mlir::impulse::finalizeWelford(), getTypes(), toValues(), and mlir::impulse::updateWelford().
| Value mlir::impulse::WelfordState::mean |
Definition at line 406 of file HMCUtils.h.
Referenced by getTypes(), toValues(), and mlir::impulse::updateWelford().
| Value mlir::impulse::WelfordState::n |
Definition at line 408 of file HMCUtils.h.
Referenced by mlir::impulse::finalizeWelford(), getTypes(), toValues(), and mlir::impulse::updateWelford().