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
New feature (non-breaking change which adds functionality)
Checklist:
[ ] My code follows the style guidelines of this project (You can use the linters)
[ ] I have performed a self-review of my own code
[ ] I have commented my code, particularly in hard-to-understand areas and hacks
[ ] I have made corresponding changes to the documentation
[ ] I have added tests to verify my fix or my feature
[ ] New and existing unit tests pass locally with my changes
[ ] I have added the relevant labels to my PR in so that relevant reviewers are notified
Description
Support
native_dropout
dynamo converter.Per the schema, this function returns two tensors. The first is
output
and the other ismask
. For inference, i.e.,train
is notTrue
orNone
, whatever thep
is, it always returnsinput
and allTrue
s.Fixes #2494
Type of change
Checklist: