TransformUtils.h File Reference

#include "Dialect/Impulse/Impulse.h"
#include "Dialect/Ops.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Math/IR/Math.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Value.h"

Include dependency graph for TransformUtils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	mlir
namespace	mlir::enzyme
namespace	mlir::enzyme::transforms

Functions
int64_t	mlir::enzyme::transforms::getUnconstrainedSize (int64_t constrainedSize, impulse::SupportKind kind)
	Get the unconstrained size given a constrained size and support kind.
int64_t	mlir::enzyme::transforms::getConstrainedSize (int64_t unconstrainedSize, impulse::SupportKind kind)
	Get the constrained size given an unconstrained size and support kind.
Value	mlir::enzyme::transforms::unconstrain (OpBuilder &builder, Location loc, Value constrained, impulse::SupportAttr support)
	Transform from constrained to unconstrained space.
Value	mlir::enzyme::transforms::constrain (OpBuilder &builder, Location loc, Value unconstrained, impulse::SupportAttr support)
	Transform from unconstrained to constrained space.
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.
Value	mlir::enzyme::transforms::createLogit (OpBuilder &builder, Location loc, Value x)
Value	mlir::enzyme::transforms::createLogSigmoid (OpBuilder &builder, Location loc, Value x)