mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy > Class Template Reference

#include "MLIR/Implementations/CoreDialectsAutoDiffImplementations.h"

Inheritance diagram for mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >:

Collaboration diagram for mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >:

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

Detailed Description

template<typename OpTy>
class mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >

Definition at line 183 of file CoreDialectsAutoDiffImplementations.h.

Member Function Documentation

cacheValues()

template<typename OpTy >

SmallVector< Value > mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >::cacheValues ( Operation * op, MGradientUtilsReverse * gutils ) const

inline

Definition at line 193 of file CoreDialectsAutoDiffImplementations.h.

createReverseModeAdjoint()

template<typename OpTy >

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

inline

Definition at line 187 of file CoreDialectsAutoDiffImplementations.h.

createShadowValues()

template<typename OpTy >

void mlir::enzyme::detail::AutoDiffUsingAllocationRev< OpTy >::createShadowValues ( Operation * op, OpBuilder & builder, MGradientUtilsReverse * gutils ) const

inline

Definition at line 198 of file CoreDialectsAutoDiffImplementations.h.

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

---

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

• MLIR/Implementations/CoreDialectsAutoDiffImplementations.h