[NNC] Make splitWithMask transform in-place (#58269)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58269
Test Plan: Imported from OSS
Reviewed By: ZolotukhinM
Differential Revision: D28427227
Pulled By: navahgar
fbshipit-source-id: 4e38a436abcf4752fd7ef6ab3666876eec6ea5ba