catalyst
catalyst copied to clipboard
Add EnumAttr to the Gradient Dialect for different differentiation methods.
Issue description
- Actual behavior: Right now, the gradient method is a StringAttr, but it would be better to have an EnumAttr.