Enzyme main
Loading...
Searching...
No Matches
mlir::impulse::NUTSContext Member List

This is the complete list of members for mlir::impulse::NUTSContext, including all inherited members.

allAddressesmlir::impulse::HMCContext
autodiffAttrsmlir::impulse::HMCContext
fnmlir::impulse::HMCContext
fnInputsmlir::impulse::HMCContext
fnResultTypesmlir::impulse::HMCContext
getElementType() constmlir::impulse::HMCContextinline
getFullTraceSize() constmlir::impulse::HMCContextinline
getPositionType() constmlir::impulse::HMCContextinline
getScalarType() constmlir::impulse::HMCContextinline
H0mlir::impulse::NUTSContext
hasConstrainedSupports() constmlir::impulse::HMCContextinline
hasCustomLogpdf() constmlir::impulse::HMCContextinline
HMCContext(FlatSymbolRefAttr fn, ArrayRef< Value > fnInputs, ArrayRef< Type > fnResultTypes, Value originalTrace, ArrayAttr selection, ArrayAttr allAddresses, Value invMass, Value massMatrixSqrt, Value stepSize, Value trajectoryLength, int64_t positionSize, ArrayRef< SupportInfo > supports, DictionaryAttr autodiffAttrs={})mlir::impulse::HMCContextinline
HMCContext(FlatSymbolRefAttr logpdfFn, ArrayRef< Value > fnInputs, Value invMass, Value massMatrixSqrt, Value stepSize, Value trajectoryLength, int64_t positionSize, DictionaryAttr autodiffAttrs={})mlir::impulse::HMCContextinline
invMassmlir::impulse::HMCContext
logpdfFnmlir::impulse::HMCContext
massMatrixSqrtmlir::impulse::HMCContext
maxDeltaEnergymlir::impulse::NUTSContext
maxTreeDepthmlir::impulse::NUTSContext
NUTSContext(FlatSymbolRefAttr fn, ArrayRef< Value > fnInputs, ArrayRef< Type > fnResultTypes, Value originalTrace, ArrayAttr selection, ArrayAttr allAddresses, Value invMass, Value massMatrixSqrt, Value stepSize, int64_t positionSize, ArrayRef< SupportInfo > supports, Value H0, Value maxDeltaEnergy, int64_t maxTreeDepth, DictionaryAttr autodiffAttrs={})mlir::impulse::NUTSContextinline
NUTSContext(FlatSymbolRefAttr logpdfFn, ArrayRef< Value > fnInputs, Value invMass, Value massMatrixSqrt, Value stepSize, int64_t positionSize, Value H0, Value maxDeltaEnergy, int64_t maxTreeDepth, DictionaryAttr autodiffAttrs={})mlir::impulse::NUTSContextinline
originalTracemlir::impulse::HMCContext
positionSizemlir::impulse::HMCContext
selectionmlir::impulse::HMCContext
stepSizemlir::impulse::HMCContext
supportsmlir::impulse::HMCContext
trajectoryLengthmlir::impulse::HMCContext
withH0(Value newH0) constmlir::impulse::NUTSContextinline
withStepSize(Value newStepSize) constmlir::impulse::HMCContextinline
Generated on Fri May 8 2026 19:56:27 for Enzyme by doxygen 1.12.0