pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.41k stars 333 forks source link

Add dynamic shape support for leaky_relu/elu/hard_sigmoid/softplus #2927

Closed lanluo-nvidia closed 2 weeks ago

lanluo-nvidia commented 2 weeks ago

Description

Add dynamic shape support for leaky_relu/elu/hard_sigmoid/softplus

Type of change

Please delete options that are not relevant and/or add your own.

Checklist: