Enable split explicit batch dim operator (#77)

Summary:
Pull Request resolved: https://github.com/pytorch/fx2trt/pull/77

ATT

Reviewed By: brad-mengchi, 842974287

Differential Revision: D36458976

fbshipit-source-id: 4bbd4025547244ba03a44af8ea2e96c4b38896a5

Author: Shirong Wu

fx/converters/acc_ops_converters.py

@@ -2578,7 +2578,7 @@ def acc_ops_masked_fill_tensor(
     return layer.get_output(0)
 
 
-@tensorrt_converter(acc_ops.split, no_explicit_batch_dim=True)
+@tensorrt_converter(acc_ops.split)
 def acc_ops_split(
     network: TRTNetwork,
     target: Target,

test/trt_lower/trt_operator_supported_test.py

@@ -31,7 +31,7 @@ def test_unsupport_node_explicit_batch_dim(self):
         class TestModule(nn.Module):
             def forward(self, x):
                 y = torch.add(input=x, other=x)
-                return torch.split(y, 2)
+                return torch.max_pool1d(y, 1)
 
         mod = TestModule()
         traced_mod = acc_tracer.trace(mod, [torch.randn(5, 2)])