xla

Add dynamism support to `aten.embedding` and `aten.split_with_sizes`

#6781

Merged