onnxruntime
565bead8 - [QNN EP] Support Softmax/LogSoftmax with any axis attribute (#17877)

Commit
2 years ago
[QNN EP] Support Softmax/LogSoftmax with any axis attribute (#17877) ### Description The QNN HTP backend only supports Softmax/LogSoftmax operators with an axis attribute set to `input_rank - 1` (i.e., the last dimension). This PR adds support for any axis by wrapping the QNN operator in transposes. ### Motivation and Context Support more models.
Parents
Loading