xla
Move device lock before the execution instead of tensor gathering
#3457
Merged

Loading