Skip to content

[Fix][Relax][ONNX] Convert OneHot depth to int - #20908

Open
tandede wants to merge 1 commit into
apache:mainfrom
tandede:candidate/tvm-20895-onehot-float-depth
Open

tandede wants to merge 1 commit into
apache:mainfrom
tandede:candidate/tvm-20895-onehot-float-depth

Conversation

@tandede

@tandede tandede commented Nov 4, 2026

Copy link
Copy Markdown
Contributor

This PR converts a constant ONNX OneHot depth value to a Python integer before passing it to Relax. ONNX permits floating-point depth tensors and specifies conversion to int64 before use.

The OneHot frontend test now covers both int64 and float depth tensors.

Fixes #20895

Testing:

  • python -m pytest -q tests/python/relax/test_frontend_onnx.py -k onehot
  • pre-commit run --files python/tvm/relax/frontend/onnx/onnx_frontend.py tests/python/relax/test_frontend_onnx.py

This branch has not been deployed

No deployments
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.

[Bug] [Relax][ONNX] OneHot does not convert floating-point depth to int64

1 participant