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 4aa4dce commit a4634f7Copy full SHA for a4634f7
py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -99,15 +99,12 @@
99
aten.nan_to_num,
100
aten.narrow,
101
# TODO: Disable the below operators once freezing is done
102
- aten.native_batch_norm,
103
aten.native_batch_norm_backward,
104
aten._native_batch_norm_legit,
105
aten._native_batch_norm_legit_functional,
106
aten._native_batch_norm_legit_no_training,
107
aten.native_dropout_backward,
108
- aten.native_group_norm,
109
aten.native_group_norm_backward,
110
- aten.native_layer_norm,
111
aten.native_layer_norm_backward,
112
aten.new_empty,
113
aten.new_full,
0 commit comments