↔ [Converter] Add support for torch.ops.aten.native_dropout.default in Torch-TensorRT
Open
Nobody has claimed this yet.
component: converters
feature request
story: Operator Coverage & Converters
- Dominant language
- Python
- Stars
- 3k
- Forks
- 410
- Avg merge
- 3d 18h
- Merged PRs (30d)
- 78
Description
[DEBUG | torch_tensorrt.dynamo.partitioning._global_partitioner]:
Unsupported or Excluded Nodes:
-
torch.ops.aten.native_dropout.default + Operator Count: 2
-
Function Schema:
-
Original PyTorch API:
-
Relevant TensorRT Documentation:
Alternatives
Additional context
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the unsupported-node report for torch.ops.aten.native_dropout.default in torch_tensorrt.dynamo.partitioning._global_partitioner. Determine the operator schema and relevant TensorRT support before identifying the converter entry point. Done means the operator is supported without appearing in the unsupported-node report.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- compilers, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100