Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[VSINPU]Fix gather OP with scalar indice issue #24061

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

xuke537
Copy link
Contributor

@xuke537 xuke537 commented Mar 17, 2025

If indices is a scalar(0 dimensional tensor) , gather OP produces incorrect output shape

Description

Motivation and Context

If indices is a scalar(0 dimensional tensor) , gather OP produces incorrect output shape
@xuke537
Copy link
Contributor Author

xuke537 commented Mar 17, 2025

@tianleiwu Could you help review and merge this patch? thanks

Indices tensor of gather op should not be the input tensor
@tianleiwu
Copy link
Contributor

/azp run Big Models, Linux Android Emulator QNN CI Pipeline, Linux CPU CI Pipeline, Linux CPU Minimal Build E2E CI Pipeline, Linux GPU CI Pipeline, Linux GPU TensorRT CI Pipeline

@tianleiwu
Copy link
Contributor

/azp run Linux OpenVINO CI Pipeline, Linux QNN CI Pipeline, ONNX Runtime Web CI Pipeline, onnxruntime-binary-size-checks-ci-pipeline, Win_TRT_Minimal_CUDA_Test_CI, Windows ARM64 QNN CI Pipeline, Windows GPU Doc Gen CI Pipeline, Windows x64 QNN CI Pipeline

Copy link

Azure Pipelines successfully started running 6 pipeline(s).

Copy link

Azure Pipelines successfully started running 8 pipeline(s).

@tianleiwu
Copy link
Contributor

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants