Refactor Argmax - Githubissues

gizatechxyz / orion

ONNX Runtime in Cairo 1.0 for verifiable ML inference using STARK

https://orion.gizatech.xyz

MIT License

163 stars 81 forks source link

Refactor Argmax #618

Closed raphaelDkhn closed 7 months ago

raphaelDkhn commented 7 months ago

Pull Request type

Please check the type of change your PR introduces:

[ ] Bugfix
[ ] Feature
[ ] Code style update (formatting, renaming)
[X] Refactoring (no functional changes, no API changes)
[ ] Build-related changes
[ ] Documentation content changes
[ ] Other (please describe):

What is the current behavior?

Refactor Argmax to fit with ONNX interface

Does this introduce a breaking change?

[X] Yes
[ ] No

Other information