mlir::impulse::NUTSTreeState Struct Reference

#include "MLIR/Interfaces/HMCUtils.h"

Collaboration diagram for mlir::impulse::NUTSTreeState:

Public Member Functions
SmallVector< Value >	toValues () const
SmallVector< Type >	getTypes () const
IntegratorState	getLeftLeaf () const
IntegratorState	getRightLeaf () const

Static Public Member Functions
static NUTSTreeState	fromValues (ArrayRef< Value > values)

Public Attributes
Value	q_left
Value	p_left
Value	grad_left
Value	q_right
Value	p_right
Value	grad_right
Value	q_proposal
Value	grad_proposal
Value	U_proposal
Value	H_proposal
Value	depth
Value	weight
Value	turning
Value	diverging
Value	sum_accept_probs
Value	num_proposals
Value	p_sum
Value	rng

Detailed Description

Definition at line 205 of file HMCUtils.h.

Member Function Documentation

fromValues()

NUTSTreeState NUTSTreeState::fromValues ( ArrayRef< Value > values )

static

Definition at line 35 of file HMCUtils.cpp.

Referenced by mlir::impulse::buildIterativeSubtree(), and mlir::impulse::buildTree().

getLeftLeaf()

IntegratorState mlir::impulse::NUTSTreeState::getLeftLeaf ( ) const

inline

Definition at line 217 of file HMCUtils.h.

References grad_left, p_left, and q_left.

getRightLeaf()

IntegratorState mlir::impulse::NUTSTreeState::getRightLeaf ( ) const

inline

Definition at line 218 of file HMCUtils.h.

References grad_right, p_right, and q_right.

getTypes()

SmallVector< Type > NUTSTreeState::getTypes

(

)

const

Definition at line 57 of file HMCUtils.cpp.

References toValues().

Referenced by mlir::impulse::buildIterativeSubtree(), and mlir::impulse::buildTree().

toValues()

SmallVector< Value > NUTSTreeState::toValues

(

)

const

Definition at line 26 of file HMCUtils.cpp.

References depth, diverging, grad_left, grad_proposal, grad_right, H_proposal, num_proposals, p_left, p_right, p_sum, q_left, q_proposal, q_right, rng, sum_accept_probs, turning, U_proposal, and weight.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::buildTree(), and getTypes().

Member Data Documentation

depth

Value mlir::impulse::NUTSTreeState::depth

Definition at line 209 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::buildTree(), mlir::impulse::combineTrees(), and toValues().

diverging

Value mlir::impulse::NUTSTreeState::diverging

Definition at line 209 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::buildTree(), mlir::impulse::combineTrees(), and toValues().

grad_left

Value mlir::impulse::NUTSTreeState::grad_left

Definition at line 206 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getLeftLeaf(), and toValues().

grad_proposal

Value mlir::impulse::NUTSTreeState::grad_proposal

Definition at line 208 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

grad_right

Value mlir::impulse::NUTSTreeState::grad_right

Definition at line 207 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getRightLeaf(), and toValues().

H_proposal

Value mlir::impulse::NUTSTreeState::H_proposal

Definition at line 208 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

num_proposals

Value mlir::impulse::NUTSTreeState::num_proposals

Definition at line 210 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::combineTrees(), and toValues().

p_left

Value mlir::impulse::NUTSTreeState::p_left

Definition at line 206 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getLeftLeaf(), and toValues().

p_right

Value mlir::impulse::NUTSTreeState::p_right

Definition at line 207 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getRightLeaf(), and toValues().

p_sum

Value mlir::impulse::NUTSTreeState::p_sum

Definition at line 210 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::combineTrees(), and toValues().

q_left

Value mlir::impulse::NUTSTreeState::q_left

Definition at line 206 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getLeftLeaf(), and toValues().

q_proposal

Value mlir::impulse::NUTSTreeState::q_proposal

Definition at line 208 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

q_right

Value mlir::impulse::NUTSTreeState::q_right

Definition at line 207 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), mlir::impulse::getLeafFromTree(), getRightLeaf(), and toValues().

rng

Value mlir::impulse::NUTSTreeState::rng

Definition at line 211 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::buildTree(), mlir::impulse::doubleTree(), and toValues().

sum_accept_probs

Value mlir::impulse::NUTSTreeState::sum_accept_probs

Definition at line 210 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

turning

Value mlir::impulse::NUTSTreeState::turning

Definition at line 209 of file HMCUtils.h.

Referenced by mlir::impulse::buildIterativeSubtree(), mlir::impulse::buildTree(), mlir::impulse::combineTrees(), and toValues().

U_proposal

Value mlir::impulse::NUTSTreeState::U_proposal

Definition at line 208 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

weight

Value mlir::impulse::NUTSTreeState::weight

Definition at line 209 of file HMCUtils.h.

Referenced by mlir::impulse::combineTrees(), and toValues().

---

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

• MLIR/Interfaces/HMCUtils.h
• MLIR/Interfaces/HMCUtils.cpp