Enabling var_mean decomposition

apbose · apbose · commit 1ad2aeba6e72 · 2023-08-28T12:33:21.000-07:00
diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -175,6 +175,7 @@
     aten.linalg_vector_norm,
     aten.full,
     aten.repeat,
+    aten.var_mean,
 }
 torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = {
     aten._softmax.default,

Original file line number	Diff line number	Diff line change
`@@ -175,6 +175,7 @@`
`175`	`175`	`aten.linalg_vector_norm,`
`176`	`176`	`aten.full,`
`177`	`177`	`aten.repeat,`
	`178`	`+ aten.var_mean,`
`178`	`179`	`}`
`179`	`180`	`torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = {`
`180`	`181`	`aten._softmax.default,`