|
Enzyme main
|
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 | |
| ForwardCachedFunctions | mlir::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 | |
| ReverseCachedFunctions | mlir::enzyme::MEnzymeLogic | |
| visitChild(Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils) | mlir::enzyme::MEnzymeLogic | |
| visitChildren(Block *oBB, Block *reverseBB, MGradientUtilsReverse *gutils) | mlir::enzyme::MEnzymeLogic |