We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e49ef6d commit 178f4f3Copy full SHA for 178f4f3
py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -125,8 +125,7 @@
125
aten.renorm,
126
aten.renorm_,
127
aten.rot90,
128
- aten.rsub.Scalar,
129
- aten.rsub.Tensor,
+ aten.rsub,
130
aten.select_backward,
131
aten.select_scatter,
132
aten.sgn,
@@ -175,6 +174,7 @@
175
174
aten.linalg_vector_norm,
176
aten.full,
177
aten.repeat,
+ aten.var_mean,
178
}
179
torch_disabled_decompositions: Set[Union[OpOverload, OpOverloadPacket]] = {
180
aten._softmax.default,
0 commit comments