Enzyme main
Loading...
Searching...
No Matches
mlir::enzyme::MEnzymeLogic Member List

This is the complete list of members for mlir::enzyme::MEnzymeLogic, including all inherited members.

CreateForwardDiff(FunctionOpInterface fn, std::vector< DIFFE_TYPE > retType, std::vector< DIFFE_TYPE > constants, MTypeAnalysis &TA, std::vector< bool > returnPrimals, DerivativeMode mode, bool freeMemory, size_t width, mlir::Type addedType, MFnTypeInfo type_args, std::vector< bool > volatile_args, void *augmented, bool omp, llvm::StringRef postpasses, bool verifyPostPasses, bool strongZero)mlir::enzyme::MEnzymeLogic
CreateReverseDiff(FunctionOpInterface fn, std::vector< DIFFE_TYPE > retType, std::vector< DIFFE_TYPE > constants, MTypeAnalysis &TA, std::vector< bool > returnPrimals, std::vector< bool > returnShadows, DerivativeMode mode, bool freeMemory, size_t width, mlir::Type addedType, MFnTypeInfo type_args, std::vector< bool > volatile_args, void *augmented, bool omp, llvm::StringRef postpasses, bool verifyPostPasses, bool strongZero)mlir::enzyme::MEnzymeLogic
differentiate(MGradientUtilsReverse *gutils, Region &oldRegion, Region &newRegion, llvm::function_ref< buildReturnFunction > buildFuncRetrunOp, std::function< std::pair< Value, Value >(Type)> cacheCreator)mlir::enzyme::MEnzymeLogic
ForwardCachedFunctionsmlir::enzyme::MEnzymeLogic
handlePredecessors(Block *oBB, Block *newBB, Block *reverseBB, MGradientUtilsReverse *gutils, llvm::function_ref< buildReturnFunction > buildReturnOp)mlir::enzyme::MEnzymeLogic
initializeShadowValues(SmallVector< mlir::Block * > &dominatorToposortBlocks, MGradientUtilsReverse *gutils)mlir::enzyme::MEnzymeLogic
mapInvertArguments(Block *oBB, Block *reverseBB, MGradientUtilsReverse *gutils)mlir::enzyme::MEnzymeLogic
ReverseCachedFunctionsmlir::enzyme::MEnzymeLogic
visitChild(Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils)mlir::enzyme::MEnzymeLogic
visitChildren(Block *oBB, Block *reverseBB, MGradientUtilsReverse *gutils)mlir::enzyme::MEnzymeLogic