mlir::enzyme::AutoDiffCallRev< OpTy > Class Template Reference

#include "MLIR/Implementations/CoreDialectsAutoDiffImplementations.h"

Inheritance diagram for mlir::enzyme::AutoDiffCallRev< OpTy >:

Collaboration diagram for mlir::enzyme::AutoDiffCallRev< OpTy >:

Public Member Functions
LogicalResult	createReverseModeAdjoint (Operation *orig, OpBuilder &builder, MGradientUtilsReverse *gutils, SmallVector< Value > caches) const
SmallVector< Value >	cacheValues (Operation *orig, MGradientUtilsReverse *gutils) const
LogicalResult	createShadowValues (Operation *op, OpBuilder &builder, MGradientUtilsReverse *gutils) const

Detailed Description

template<typename OpTy>
class mlir::enzyme::AutoDiffCallRev< OpTy >

Definition at line 239 of file CoreDialectsAutoDiffImplementations.h.

Member Function Documentation

cacheValues()

template<typename OpTy >

SmallVector< Value > mlir::enzyme::AutoDiffCallRev< OpTy >::cacheValues ( Operation * orig, MGradientUtilsReverse * gutils ) const

inline

Definition at line 249 of file CoreDialectsAutoDiffImplementations.h.

References mlir::enzyme::detail::callCacheValues().

createReverseModeAdjoint()

template<typename OpTy >

LogicalResult mlir::enzyme::AutoDiffCallRev< OpTy >::createReverseModeAdjoint ( Operation * orig, OpBuilder & builder, MGradientUtilsReverse * gutils, SmallVector< Value > caches ) const

inline

Definition at line 243 of file CoreDialectsAutoDiffImplementations.h.

References mlir::enzyme::detail::callReverseHandler().

createShadowValues()

template<typename OpTy >

LogicalResult mlir::enzyme::AutoDiffCallRev< OpTy >::createShadowValues ( Operation * op, OpBuilder & builder, MGradientUtilsReverse * gutils ) const

inline

Definition at line 254 of file CoreDialectsAutoDiffImplementations.h.

---

The documentation for this class was generated from the following file:

• MLIR/Implementations/CoreDialectsAutoDiffImplementations.h