torch-mlir icon indicating copy to clipboard operation
torch-mlir copied to clipboard

[onnx] Decompose `onnx.Adam` into elementwise operations

Open rsuderman opened this issue 1 year ago • 0 comments

We can support onnx.Adam by decomposing into elementwise operations. The numerics slightly differ from onnxruntime however they appear to be within margins.

rsuderman avatar Aug 16 '24 21:08 rsuderman