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

feat: support `native_dropout` dynamo converter #2931

Closed zewenli98 closed 2 weeks ago

zewenli98 commented 2 weeks ago

Description

Support native_dropout dynamo converter.

Per the schema, this function returns two tensors. The first is output and the other is mask. For inference, i.e., train is not True or None, whatever the p is, it always returns input and all Trues.

Fixes #2494

Type of change

Checklist: