onnxruntime
[QNN EP] Support Softmax/LogSoftmax with any axis attribute
#17877

Merged

[QNN EP] Support Softmax/LogSoftmax with any axis attribute #17877

adrianlizarraga merged 5 commits into main from adrianl/qnn-softmax-axis-support

Use topological order traversal when validating graph (same as when c…

5f137ae5

Add separate op builder for Softmax/LogSoftmax

bf83f04f

Support Softmax/LogSoftmax (opset > 13) with arbitrary axis attributes

e3b27a2b

adrianlizarraga commented on 2023-10-11

adrianlizarraga marked this pull request as ready for review 2 years ago

adrianlizarraga requested a review from

HectorSVC 2 years ago

adrianlizarraga requested a review from

jywu-msft 2 years ago

Fix bug in output transpose tensor type. QNN CPU backend only support…

df6403b1

HectorSVC commented on 2023-10-11

Do not support Softmax with initializer input

0012ba3f

HectorSVC approved these changes on 2023-10-12

adrianlizarraga merged 565bead8 into main 2 years ago

adrianlizarraga deleted the adrianl/qnn-softmax-axis-support branch 2 years ago

Reviewers

HectorSVC

jywu-msft

Assignees

No one assigned

Labels

None yet

Milestone

No milestone