File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed
py/torch_tensorrt/dynamo/conversion Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -57,6 +57,7 @@ def one_user_validator(node: Node) -> bool:
57
57
58
58
@dynamo_tensorrt_converter (torch .ops .aten .native_batch_norm .default , capability_validator = one_user_validator ) # type: ignore[misc]
59
59
@dynamo_tensorrt_converter (torch .ops .aten .batch_norm .default ) # type: ignore[misc]
60
+ @dynamo_tensorrt_converter (torch .ops .aten .batch_norm ) # type: ignore[misc]
60
61
@enforce_tensor_types (
61
62
{
62
63
0 : (TRTTensor ,),
@@ -89,6 +90,7 @@ def aten_ops_batch_norm(
89
90
90
91
@dynamo_tensorrt_converter (torch .ops .aten .native_layer_norm .default , capability_validator = one_user_validator ) # type: ignore[misc]
91
92
@dynamo_tensorrt_converter (torch .ops .aten .layer_norm .default ) # type: ignore[misc]
93
+ @dynamo_tensorrt_converter (torch .ops .aten .layer_norm ) # type: ignore[misc]
92
94
@enforce_tensor_types (
93
95
{
94
96
0 : (TRTTensor ,),
@@ -146,6 +148,7 @@ def aten_ops_native_group_norm(
146
148
147
149
148
150
@dynamo_tensorrt_converter (torch .ops .aten .group_norm .default ) # type: ignore[misc]
151
+ @dynamo_tensorrt_converter (torch .ops .aten .group_norm ) # type: ignore[misc]
149
152
@enforce_tensor_types (
150
153
{
151
154
0 : (TRTTensor ,),
You can’t perform that action at this time.
0 commit comments