Skip to content

Update TensorRT dynamic shape profile when input shape changed during runtime#3904

Merged
stevenlix merged 3 commits intomasterfrom
stevenlix/onestage
May 11, 2020
Merged

Update TensorRT dynamic shape profile when input shape changed during runtime#3904
stevenlix merged 3 commits intomasterfrom
stevenlix/onestage

Conversation

@stevenlix
Copy link
Contributor

  • TensorRT EP needs to create new engine if input shape changed during runtime and the new shape is not covered by current engine. This PR updates TensorRT dynamic shape profile based on the new shape and creates engine accordingly.

@stevenlix stevenlix requested a review from jywu-msft May 11, 2020 20:07
@stevenlix stevenlix requested a review from a team as a code owner May 11, 2020 20:07
@stevenlix stevenlix merged commit 28f693a into master May 11, 2020
@stevenlix stevenlix deleted the stevenlix/onestage branch May 11, 2020 23:11
stevenlix added a commit that referenced this pull request May 12, 2020
… runtime (#3904)

* Update dynamic shape range when input shape changed during runtime

* Update tensorrt_execution_provider.cc

* Update tensorrt_execution_provider.cc
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