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

This is the complete list of members for mlir::impulse::HMCContext, 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
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
originalTracemlir::impulse::HMCContext
positionSizemlir::impulse::HMCContext
selectionmlir::impulse::HMCContext
stepSizemlir::impulse::HMCContext
supportsmlir::impulse::HMCContext
trajectoryLengthmlir::impulse::HMCContext
withStepSize(Value newStepSize) constmlir::impulse::HMCContextinline