Revert "Revert "Hold WeakReference to split callback""

This reverts commit 06be7eb88ee96d1eba0bcedc7728d9f64c2bd6d1.

Reason for revert: Potential forward fix with ag/24562796

Change-Id: I1f01dda728491e627ec0cc34fd364cbd794854d5
1 file changed