pytorch
71587e05 - modify the check condition of Conv-> Add/Sub/Mul/Div folding

Commit
2 years ago
modify the check condition of Conv-> Add/Sub/Mul/Div folding Relax the check condition of Conv-> Add/Sub/Mul/Div folding to accept that the input tensor of Add/Sub/Mul/Div is floating type or the promoteTypes of the input tensor of Add/Sub/Mul/Div is equal to the type of conv weight. Relaxing this condition is mainly to deal with a common situation in models: the conv output add/sub/mul/div an integer tensor or integer scalar tensor. Pull Request resolved: https://github.com/pytorch/pytorch/pull/73278 Approved by: https://github.com/eellison
Author
Committer
Parents
Loading