mlir::enzyme::transforms Namespace Reference

Functions
int64_t	getUnconstrainedSize (int64_t constrainedSize, impulse::SupportKind kind)
	Get the unconstrained size given a constrained size and support kind.
int64_t	getConstrainedSize (int64_t unconstrainedSize, impulse::SupportKind kind)
	Get the constrained size given an unconstrained size and support kind.
Value	unconstrain (OpBuilder &builder, Location loc, Value constrained, impulse::SupportAttr support)
	Transform from constrained to unconstrained space.
Value	constrain (OpBuilder &builder, Location loc, Value unconstrained, impulse::SupportAttr support)
	Transform from unconstrained to constrained space.
Value	logAbsDetJacobian (OpBuilder &builder, Location loc, Value unconstrained, impulse::SupportAttr support)
	Compute log |det J| of the transform from unconstrained to constrained.
Value	createLogit (OpBuilder &builder, Location loc, Value x)
Value	createLogSigmoid (OpBuilder &builder, Location loc, Value x)

Function Documentation

constrain()

Value mlir::enzyme::transforms::constrain	(	OpBuilder &	builder,
		Location	loc,
		Value	unconstrained,
		impulse::SupportAttr	support )

Transform from unconstrained to constrained space.

Definition at line 146 of file TransformUtils.cpp.

Referenced by mlir::impulse::constrainPosition().

createLogit()

Value mlir::enzyme::transforms::createLogit	(	OpBuilder &	builder,
		Location	loc,
		Value	x )

Definition at line 23 of file TransformUtils.cpp.

Referenced by unconstrain().

createLogSigmoid()

Value mlir::enzyme::transforms::createLogSigmoid	(	OpBuilder &	builder,
		Location	loc,
		Value	x )

Definition at line 36 of file TransformUtils.cpp.

Referenced by logAbsDetJacobian().

getConstrainedSize()

int64_t mlir::enzyme::transforms::getConstrainedSize	(	int64_t	unconstrainedSize,
		impulse::SupportKind	kind )

Get the constrained size given an unconstrained size and support kind.

Definition at line 64 of file TransformUtils.cpp.

getUnconstrainedSize()

int64_t mlir::enzyme::transforms::getUnconstrainedSize	(	int64_t	constrainedSize,
		impulse::SupportKind	kind )

Get the unconstrained size given a constrained size and support kind.

Definition at line 50 of file TransformUtils.cpp.

logAbsDetJacobian()

Value mlir::enzyme::transforms::logAbsDetJacobian	(	OpBuilder &	builder,
		Location	loc,
		Value	unconstrained,
		impulse::SupportAttr	support )

Compute log |det J| of the transform from unconstrained to constrained.

Definition at line 218 of file TransformUtils.cpp.

References createLogSigmoid().

Referenced by mlir::impulse::computeTotalJacobianCorrection().

unconstrain()

Value mlir::enzyme::transforms::unconstrain	(	OpBuilder &	builder,
		Location	loc,
		Value	constrained,
		impulse::SupportAttr	support )

Transform from constrained to unconstrained space.

Definition at line 78 of file TransformUtils.cpp.

References createLogit().

Referenced by mlir::impulse::unconstrainPosition().