pytorch
ebf45a07 - [NNC] support aten::_convolution when it is 2D conv (#84038)

Commit
2 years ago
[NNC] support aten::_convolution when it is 2D conv (#84038) ## Motivation Currently, only `aten::conv2d` has been supported in NNC. When using `torch.jit.trace`, the node on the graph will be `aten::_convolution`. This PR adds support of `aten::_convolution` node when it corresponds to a 2D convolution. ## Pitch Support `aten::_convolution` in NNC when we can infer from the parameters that it is a 2D convolution to support models obtained from `torch.jit.trace`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/84038 Approved by: https://github.com/huiguoo
Author
Committer
Parents
Loading