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

#include "MLIR/Implementations/CoreDialectsAutoDiffImplementations.h"

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

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

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

Detailed Description

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

Definition at line 135 of file CoreDialectsAutoDiffImplementations.h.

Member Function Documentation

cacheValues()

template<typename OpTy >

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

inline

Definition at line 146 of file CoreDialectsAutoDiffImplementations.h.

createReverseModeAdjoint()

template<typename OpTy >

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

inline

Definition at line 139 of file CoreDialectsAutoDiffImplementations.h.

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

createShadowValues()

template<typename OpTy >

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

inline

Definition at line 151 of file CoreDialectsAutoDiffImplementations.h.

---

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

• MLIR/Implementations/CoreDialectsAutoDiffImplementations.h