Skip to content

Commit 794e2bb

Browse files
committed
update decomposition_groups
1 parent 426c2f4 commit 794e2bb

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,15 +99,12 @@
9999
aten.nan_to_num,
100100
aten.narrow,
101101
# TODO: Disable the below operators once freezing is done
102-
aten.native_batch_norm,
103102
aten.native_batch_norm_backward,
104103
aten._native_batch_norm_legit,
105104
aten._native_batch_norm_legit_functional,
106105
aten._native_batch_norm_legit_no_training,
107106
aten.native_dropout_backward,
108-
aten.native_group_norm,
109107
aten.native_group_norm_backward,
110-
aten.native_layer_norm,
111108
aten.native_layer_norm_backward,
112109
aten.new_empty,
113110
aten.new_full,

0 commit comments

Comments
 (0)