onnxruntime
f390eb5a - [QNN-EP] Support non-last axis TopK. (#24881)

Commit
297 days ago
[QNN-EP] Support non-last axis TopK. (#24881) ### Description In TopK op builder, add Transpose around TopK to permute the axis to the last before and permute back after. Additionally, since TopK's second output is indices which may have INT64 dtype, add Cast to cast transformed INT32 back to INT64 if is graph output. ### Motivation and Context QNN only accepts TopK on the last axis but ONNX/ORT's TopK has axis attribute. Complement TopK op builder to avoid falling back to CPU for non-last axis TopK.
Author
Parents
Loading