Temporarily add aten.op to supported quant modules, Commit 1

Differential Revision: https://internalfb.com/D48488927

fbshipit-source-id: 496fd4244abb273d8f42216e146126e2034e34c7

Author: mcr229

backends/xnnpack/partition/configs.py

@@ -111,6 +111,10 @@
     torch.nn.functional.leaky_relu,
     torch.nn.functional.leaky_relu_,
     torch.nn.LeakyReLU,
+    # TODO(): In quant --> export flow source_fn is operator target instead of module name
+    # This is actively being fixed, but until, we add these operator target names to partitioenr
+    torch.ops.aten.convolution.default,
+    torch.ops.aten.addmm.default,
 ]
 
 SUPPORTED_IMPLICIT_Q_DQ_MODULES_SET = set(SUPPORTED_QUANT_MODULES)

backends/xnnpack/passes/convert_to_linear.py

@@ -27,6 +27,7 @@ class ConvertToLinearPass(ExportPass):
     linear_modules = [
         torch.nn.Linear,
         torch.nn.functional.linear,
+        torch.ops.aten.addmm.default,
     ]
 
     targets = [