Searched refs:allow_cross_device (Results 1 – 1 of 1) sorted by relevance
1950 def copy_to(a: Tensor, b: Tensor, *, allow_cross_device=True): argument1951 if not allow_cross_device and a.device != b.device: