Michael Yang
45f56355d5
feat: uneven splits ( #11048 )
...
The current splitDim function only operates on tensors that are split evenly which isn't always the case, e.g. a QKV tensor. This change allows the function to be used for arbitrary splits
2025-06-11 12:10:54 -07:00
..
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-15 12:15:01 -07:00
2025-05-06 11:20:48 -07:00
2025-05-06 11:20:48 -07:00
2025-05-22 11:30:49 -07:00
2025-05-06 11:20:48 -07:00
2025-05-13 20:58:02 -07:00
2025-06-11 12:10:54 -07:00
2025-05-19 09:54:22 -07:00
2025-05-16 13:40:23 -07:00
2025-04-25 16:59:20 -07:00
2025-04-25 16:59:20 -07:00
2025-05-13 17:36:02 -07:00
2025-06-11 12:10:54 -07:00
2025-06-11 12:10:54 -07:00
2025-05-16 13:40:23 -07:00
2025-05-16 13:40:23 -07:00