[COREF] en_coreference_web_trf(3.4.0a2) breaks by storing some tensors on CPU and some on GPU

[COREF] en_coreference_web_trf(3.4.0a2) breaks by storing some tensors on CPU and some on GPU #15776