Merge pull request #1837 from gs-olive/_native_batch_norm_legit_no_training_fix

fix: Remove references to `_native_batch_norm_legit_no_training` for PyTorch 2.0 Stable

Author: peri044

py/torch_tensorrt/fx/passes/lower_basic_pass_aten.py

@@ -165,7 +165,6 @@ def replace_aten_op_with_indices(module: torch.fx.GraphModule) -> torch.fx.Graph
             torch.ops.aten.max_pool3d_with_indices.default,
             torch.ops.aten.native_batch_norm.default,
             torch.ops.aten._native_batch_norm_legit.default,
-            torch.ops.aten._native_batch_norm_legit_no_training.default,
         ):
             modified = True
             if len(n.users) != 1:
@@ -186,16 +185,6 @@ def replace_aten_op_with_indices(module: torch.fx.GraphModule) -> torch.fx.Graph
                 new_args = list(n.args)
                 new_args.append(False)
                 new_args = tuple(new_args)
-            elif (
-                n.target == torch.ops.aten._native_batch_norm_legit_no_training.default
-            ):
-                new_op = torch.ops.aten.batch_norm
-                new_args = list(n.args)
-                new_args.append(False)
-                # _native_batch_norm_legit_no_training doesn't take in a training arg (assumed to be false)
-                # but batchnorm takes in a training arg at position 5.
-                new_args.insert(5, False)
-                new_args = tuple(new_args)
 
             getitem_node = next(iter(n.users))
             with module.graph.inserting_after(getitem_node):