pytorch
a7406516 - Refactor bias and weight check and add aten::linear pattern (#30474)

Commit
4 years ago
Refactor bias and weight check and add aten::linear pattern (#30474) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30474 There are some common parts in `isBiasOfConvOrLinear` and `isWeightOfConvOrLinear`, we can factor them out, the refactor will allow for easier extension of new patterns Test Plan: python test/test_jit.py python test/test_quantization.py Imported from OSS Differential Revision: D18795725 fbshipit-source-id: 446463da5e3fa8464db441ed0d9651930487b3b7
Author
Parents
Loading