xla

05e947a3 - [Functionalization] detach_copy use set_ instead of _copy_from (#4785)

[Functionalization] detach_copy use set_ instead of _copy_from (#4785) Summary: It looks like _copy_from somehow is not working correctly with PJRT on devices other than xla:0. Let's use set_ which also more loyally represents the semantics of detach. Test Plan: PJRT_DEVICE=TPU python test/pjrt/test_experimental_pjrt_tpu.py TestTpuCollectiveOps.test_broadcast_master_param_synchronized_parameters