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

#include "MLIR/Interfaces/HMCUtils.h"

Collaboration diagram for mlir::impulse::DualAveragingState:

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
 

Detailed Description

Definition at line 71 of file HMCUtils.h.

Member Function Documentation

◆ fromValues()

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

Definition at line 82 of file HMCUtils.h.

◆ getTypes()

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

Definition at line 85 of file HMCUtils.h.

References gradient_avg, log_step_size, log_step_size_avg, prox_center, and step_count.

◆ toValues()

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

Definition at line 78 of file HMCUtils.h.

References gradient_avg, log_step_size, log_step_size_avg, prox_center, and step_count.

Member Data Documentation

◆ gradient_avg

Value mlir::impulse::DualAveragingState::gradient_avg

Definition at line 74 of file HMCUtils.h.

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

◆ log_step_size

Value mlir::impulse::DualAveragingState::log_step_size

Definition at line 72 of file HMCUtils.h.

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

◆ log_step_size_avg

Value mlir::impulse::DualAveragingState::log_step_size_avg

◆ prox_center

Value mlir::impulse::DualAveragingState::prox_center

Definition at line 76 of file HMCUtils.h.

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

◆ step_count

Value mlir::impulse::DualAveragingState::step_count

Definition at line 75 of file HMCUtils.h.

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


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