Commit Graph

2 Commits

Author SHA1 Message Date
Michael Yang 69ee842b6e
convert: utility for merging tensors (#11069) 2025-12-29 06:38:17 -06:00
Michael Yang db3a312edf
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-12-29 06:38:14 -06:00