(//core): Added a variant for aten::split

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

core/conversion/converters/impl/select.cpp

@@ -344,6 +344,12 @@ auto select_registrations TORCHTRT_UNUSED =
                     LOG_DEBUG("Converted split op into a list of IValues");
                     return true;
                   }})
+        .pattern({"aten::split.sizes(Tensor(a -> *) self, int[] split_size, int dim=0) -> (Tensor[])",
+                  [](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
+                    add_split(ctx, n, args, true);
+                    LOG_DEBUG("Converted split op into a list of IValues");
+                    return true;
+                  }})
         .pattern({"aten::split.Tensor(Tensor(a) self, int split_size, int dim=0) -> (Tensor[])",
                   [](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
                     add_split(ctx, n, args, false);