0 0 1 Python

Cleanup log_softmax & softmax (#791)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/791

Refactors helper functions for checking arguments, to be reused by optimized `log_softmax`
ghstack-source-id: 203560636
exported-using-ghexport

Reviewed By: cbilgin

Differential Revision: D50130021

fbshipit-source-id: f964002a195563aef7d1b573b453da5423585647

Manuel Candales committed 2y ago

e89b744004f451e206fbd5d436d4f834dd7b047a

Parent: 20df1a2

Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 10/10/2023, 9:19:39 PM