search

cyrusbehr / tensorrt-cpp-api

TensorRT C++ API Tutorial
MIT License
612 stars 76 forks source link

Support for Dynamic Width of Inputs (e.g., LSTM Models) #81

Closed thomaskleiven closed 2 months ago

thomaskleiven commented 2 months ago

It would be great to add support for dynamic width of inputs, as certain models, such as LSTMs, require this flexibility. Currently, the API is limited to fixed input dimensions, which restricts the implementation of models that adapt to varying sequence lengths.

cyrusbehr commented 2 months ago

https://github.com/cyrusbehr/tensorrt-cpp-api/pull/82

Merged

cyrusbehr commented 2 months ago

Thanks for the great work!