tensor_split -> chunk in apply_rot_embed_cat .. Fix #2594 coreml export

Author: rwightman

timm/layers/pos_embed_sincos.py

@@ -273,7 +273,7 @@ def apply_rot_embed_cat(
         emb: torch.Tensor,
         half: bool = False
 ) -> torch.Tensor:
-    sin_emb, cos_emb = emb.tensor_split(2, -1)
+    sin_emb, cos_emb = emb.chunk(2, -1)
     # x: [..., D], eg [x0, x1, x2, x3, x4, x5]
     if half:
         # sin: [..., D], eg [sin0, sin1, sin2, sin0, sin1, sin2]