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

Global tensors with dynamic slice using python #4229

Open
zengrh3 opened this issue Oct 30, 2024 · 1 comment
Open

Global tensors with dynamic slice using python #4229

zengrh3 opened this issue Oct 30, 2024 · 1 comment
Labels
question Further information is requested triaged Issue has been triaged by maintainers

Comments

@zengrh3
Copy link

zengrh3 commented Oct 30, 2024

Let say I define a global tensor t1, where shape is [1, 12, 2000, 112], and the actual input tensor t2 shape is [1, 12, 40, 112]. I have copied the actual value of t2 to t1 using _copy, and in the inner model logic, I use slice to get the actual content, like t1[:, :, :40, :].

However, I found this does not work in TensorRT, but works in ONNX. So how can I deal with this ?

@yuanyao-nv yuanyao-nv added question Further information is requested triaged Issue has been triaged by maintainers labels Oct 31, 2024
@yuanyao-nv
Copy link
Collaborator

Are you able to create an ONNX model with this? You can try parsing an ONNX model with TensorRT https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#onnx-intro

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested triaged Issue has been triaged by maintainers
Projects
None yet
Development

No branches or pull requests

2 participants