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:
5757
5858@dynamo_tensorrt_converter (torch .ops .aten .native_batch_norm .default , capability_validator = one_user_validator ) # type: ignore[misc]
5959@dynamo_tensorrt_converter (torch .ops .aten .batch_norm .default ) # type: ignore[misc]
60+ @dynamo_tensorrt_converter (torch .ops .aten .batch_norm ) # type: ignore[misc]
6061@enforce_tensor_types (
6162 {
6263 0 : (TRTTensor ,),
@@ -89,6 +90,7 @@ def aten_ops_batch_norm(
8990
9091@dynamo_tensorrt_converter (torch .ops .aten .native_layer_norm .default , capability_validator = one_user_validator ) # type: ignore[misc]
9192@dynamo_tensorrt_converter (torch .ops .aten .layer_norm .default ) # type: ignore[misc]
93+ @dynamo_tensorrt_converter (torch .ops .aten .layer_norm ) # type: ignore[misc]
9294@enforce_tensor_types (
9395 {
9496 0 : (TRTTensor ,),
@@ -146,6 +148,7 @@ def aten_ops_native_group_norm(
146148
147149
148150@dynamo_tensorrt_converter (torch .ops .aten .group_norm .default ) # type: ignore[misc]
151+ @dynamo_tensorrt_converter (torch .ops .aten .group_norm ) # type: ignore[misc]
149152@enforce_tensor_types (
150153 {
151154 0 : (TRTTensor ,),
You can’t perform that action at this time.
0 commit comments