|
Enzyme main
|
#include "MLIR/Interfaces/HMCUtils.h"

Public Member Functions | |
| SmallVector< Value > | toValues () const |
| SmallVector< Type > | getTypes () const |
Static Public Member Functions | |
| static DualAveragingState | fromValues (ArrayRef< Value > values) |
Public Attributes | |
| Value | log_step_size |
| Value | log_step_size_avg |
| Value | gradient_avg |
| Value | step_count |
| Value | prox_center |
Definition at line 71 of file HMCUtils.h.
|
inlinestatic |
Definition at line 82 of file HMCUtils.h.
|
inline |
Definition at line 85 of file HMCUtils.h.
References gradient_avg, log_step_size, log_step_size_avg, prox_center, and step_count.
|
inline |
Definition at line 78 of file HMCUtils.h.
References gradient_avg, log_step_size, log_step_size_avg, prox_center, and step_count.
| Value mlir::impulse::DualAveragingState::gradient_avg |
Definition at line 74 of file HMCUtils.h.
Referenced by getTypes(), toValues(), and mlir::impulse::updateDualAveraging().
| Value mlir::impulse::DualAveragingState::log_step_size |
Definition at line 72 of file HMCUtils.h.
Referenced by mlir::impulse::getStepSizeFromDualAveraging(), getTypes(), and toValues().
| Value mlir::impulse::DualAveragingState::log_step_size_avg |
Definition at line 73 of file HMCUtils.h.
Referenced by mlir::impulse::getStepSizeFromDualAveraging(), getTypes(), toValues(), and mlir::impulse::updateDualAveraging().
| Value mlir::impulse::DualAveragingState::prox_center |
Definition at line 76 of file HMCUtils.h.
Referenced by getTypes(), toValues(), and mlir::impulse::updateDualAveraging().
| Value mlir::impulse::DualAveragingState::step_count |
Definition at line 75 of file HMCUtils.h.
Referenced by getTypes(), toValues(), and mlir::impulse::updateDualAveraging().