pytorch
9ad14f6b - cover nn.Conv1d in mkldnn model conversion logic (#38528)

Commit
4 years ago
cover nn.Conv1d in mkldnn model conversion logic (#38528) Summary: current `to_mkldnn` model conversion logic under `torch.utils.mkldnn` does not cover `nn.Conv1d`. This patch fills the gap, using similar logic to `nn.Conv2d`. The model conversion will remove unnecessary memory format reorders of input/output tensors and thus speedup the model. Pull Request resolved: https://github.com/pytorch/pytorch/pull/38528 Differential Revision: D21640325 Pulled By: albanD fbshipit-source-id: c3340153b5c524e020c097eb4b9e2ffcbde8896d
Author
Parents
Loading